【堆 贪心】 502 IPO

题目

资源有限，它只能在 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;

}