贪心算法-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;
   }
}