贪心算法-IPO问题
1、题目描述
给你一个启动资金w,和一个最大项目次数k。
然后,有两个数组,一个cost[],里面记录了每个项目需要花费的资金。一个profit数组,里面记录了每个项目完成后可以获取的利润。然后请你计算出,给你一个初始资金w,和最大项目次数k的情况下,可以获取的最大利润。每次只能做一个项目,不能同时进行几个项目。
例如:初始资金10,k=3,cost[10,20,30,40],profit[10,20,30,40]
那么,初始资金只能够花费来进行第一个10的项目,其他项目买不起,然后,第一个项目做完,利润是10,现在手里有20,就可以进行第二个项目了,然后再累加利润,看看能不能再解锁新项目。注意,每个项目只能做一次。
2、思路解析
升级打怪思路,就是我来到新手村,我肯定是找自己能干的过的,并且对我能力值提升高的怪物来打
就是搞两个堆小根堆+大根堆
小根堆里放所有项目 按项目花费资金排序
大根堆里放所有当前资金能覆盖的项目,按利润排序
最后把大根堆弹出的项目的利润加在一起即为所得
代码实现
// 最多K个项目
// W是初始资金
// Profits[] Capital[] 一定等长
// 返回最终最大的资金
public static int findMaximizedCapital(int K, int W, int[] Profits, int[] Capital) {
PriorityQueue<Program> minCostQ = new PriorityQueue<>(new MinCostComparator());
PriorityQueue<Program> maxProfitQ = new PriorityQueue<>(new MaxProfitComparator());
for (int i = 0; i < Profits.length; i++) {
minCostQ.add(new Program(Profits[i], Capital[i]));
}
for (int i = 0; i < K; i++) {
while (!minCostQ.isEmpty() && minCostQ.peek().c <= W) {
maxProfitQ.add(minCostQ.poll());
}
//可能资本不足,造成现在还能做项目但是大根堆没有项目给你做了
if (maxProfitQ.isEmpty()) {
return W;
}
W += maxProfitQ.poll().p;
}
return W;
}
public static class Program {
public int p;
public int c;
public Program(int p, int c) {
this.p = p;
this.c = c;
}
}
public static class MinCostComparator implements Comparator<Program> {
@Override
public int compare(Program o1, Program o2) {
return o1.c - o2.c;
}
}
public static class MaxProfitComparator implements Comparator<Program> {
@Override
public int compare(Program o1, Program o2) {
return o2.p - o1.p;
}
}