1. // 项目 花费 和 利润
    2.     public static class Program {
    3.         public int p;
    4.         public int c;
    6.         public Program(int p, int c) {
    7.             this.p = p;
    8.             this.c = c;
    9.         }
    10.     }
    12.     // 根据花费组织的小根堆比较器
    13.     public static class MinCostComparator implements Comparator<Program> {
    14.         @Override
    15.         public int compare(Program o1, Program o2) {
    16.             return o1.c - o2.c;
    17.         }
    18.     }
    20.     // 根据利润组织的大根堆比较器
    21.     public static class MaxProfitComparator implements Comparator<Program> {
    22.         @Override
    23.         public int compare(Program o1, Program o2) {
    24.             return o2.p - o1.p;
    25.         }
    27.     }
    28.     // 最多K个项目
    29.     // W是初始资金
    30.     // Profits[] Capital[] 一定等长
    31.     // 返回最终最大的资金
    32.     public static int findMaximizedCapital(int K, int W, int[] Profits, int[] Capital) {
    33.         PriorityQueue<Program> minCostQ = new PriorityQueue<>(new MinCostComparator());
    34.         PriorityQueue<Program> maxProfitQ = new PriorityQueue<>(new MaxProfitComparator());
    36.         // 先把所有的项目放入小根堆
    37.         for (int i = 0; i < Profits.length; i++) {
    38.             minCostQ.add(new Program(Profits[i], Capital[i]));
    39.         }
    41.         for (int i = 0; i < K; i++) { // 进行 k 轮
    42.             //peek()获取堆顶元素但不删除
    43.             // 力所能及的项目，全解锁
    44.             while (!minCostQ.isEmpty() && minCostQ.peek().c <= W) {
    45.                 // 放入大根堆中
    46.                 maxProfitQ.add(minCostQ.poll());
    47.             }
    49.             // 没有进行k轮，但是没有能做的项目了，直接返回
    50.             if (maxProfitQ.isEmpty()) {
    51.                 return W;
    52.             }
    54.             // 累加利润
    55.             W += maxProfitQ.poll().p;
    56.         }
    57.         return W;
    58.     }