题目
资源有限,它只能在 IPO 之前完成最多 k 个不同的项目。帮助 力扣 设计完成最多 k 个不同项目后得到最大总资本的方式。
给定若干个项目。对于每个项目 i,它都有一个纯利润 Pi,并且需要最小的资本 Ci 来启动相应的项目。最初,你有 W 资本。当你完成一个项目时,你将获得纯利润,且利润将被添加到你的总资本中。
总而言之,从给定项目中选择最多 k 个不同项目的列表,以最大化最终资本,并输出最终可获得的最多资本。
输入: k=2, W=0, Profits=[1,2,3], Capital=[0,1,1].
输出: 4
思路
建立一个类型,储存资本和收益。建立一个数组,数组元素是每个项目,每个项目有两个属性是资本和收益。
按照资本建立小跟堆,当堆顶小于当前资本数时,弹出数组到大根堆中,直到无法弹出,大根堆按照收益建立,弹出堆顶,将收益累加到资本中。再看此时小跟堆能否弹出。最后大根堆为空时返回此时的资本。
代码
public class Node{
public int p;
public int c;
public Node(int p,int c){
this.p = p;
this.c = c;
}
}
//定义小跟堆比较器
public class MinCostComparator implements Comparator<Node>{
@Override
public int compare(Node o1, Node o2) {
return o1.c-o2.c;
}
}
//定义大根堆比较器
public class MaxProfitComparator implements Comparator<Node>{
@Override
public int compare(Node o1, Node o2) {
return o2.p-o1.p;
}
}
public int findMaximizedCapital(int k, int W, int[] Profits, int[] Capital) {
Node[] nodes = new Node[Profits.length];
//项目生成
for (int i = 0; i < Profits.length; i++) {
nodes[i] = new Node(Profits[i],Capital[i]);
}
//按照资本建立小跟堆
PriorityQueue<Node> minCostQ = new PriorityQueue<>(new MinCostComparator());
//按照利益建立大根堆
PriorityQueue<Node> maxProfitQ = new PriorityQueue<>(new MaxProfitComparator());
for (int i = 0; i < nodes.length; i++) {
minCostQ.add(nodes[i]);
}
for (int i = 0; i < k; i++) {
//小跟堆堆顶小于k时就弹出加入到大根堆
while(!minCostQ.isEmpty() && minCostQ.peek().c<=W){
maxProfitQ.add(minCostQ.poll());
}
//大根堆空时,返回当前收益值
if(maxProfitQ.isEmpty()){
return W;
}
//大根堆不空时,弹出收益最大的值,同时收益累加,累加后可以在小跟堆中弹出新的项目
W+=maxProfitQ.poll().p;
}
return W;
}