LeetCode 502. IPO (贪心算法)

思路

代码实现

class Solution {
    public int findMaximizedCapital(int k, int w, int[] profits, int[] capital) {
        // 准备一个小根堆，按照花费排序
        PriorityQueue<Program> minCostQ = new PriorityQueue<>((Program p1, Program p2) -> p1.c - p2.c);
        // 准备一个大根堆，按照利润排序
        PriorityQueue<Program> maxProfitQ = new PriorityQueue<>((Program p1, Program p2) -> p2.p - p1.p);

        // 先将元素封装到Program中加入到小根堆
        for(int i = 0; i < capital.length; i++) {
            minCostQ.add(new Program(profits[i], capital[i]));
        }

        // 遍历K次
        for(int i = 0; i < k; i++) {
            // 如果小根堆不为空，并且小根堆中项目的花费小于等于现有资金，则说明可以作为项目备选项，加入大根堆
            while(!minCostQ.isEmpty() && w >= minCostQ.peek().c) {
                maxProfitQ.add(minCostQ.poll());
            }
            // 如果当前已经没有项目了，则直接返回现有利润
            if(maxProfitQ.isEmpty()) return w;
            // 还有项目，则选择利润最大的可做项目，将利润累加
            w += maxProfitQ.poll().p;
        }
        return w;
    }

    public class Program {
		public int p;
		public int c;

		public Program(int p, int c) {
			this.p = p;
			this.c = c;
		}
	}
}