贪心算法-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、思路解析

升级打怪思路，就是我来到新手村，我肯定是找自己能干的过的，并且对我能力值提升高的怪物来打

就是搞两个堆小根堆+大根堆

小根堆里放所有项目按项目花费资金排序

大根堆里放所有当前资金能覆盖的项目，按利润排序

最后把大根堆弹出的项目的利润加在一起即为所得

代码实现

java 复制代码

// 最多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;
   }
}