贪心算法-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;
   }
}
相关推荐
莹莹学编程—成长记1 小时前
string的模拟实现
服务器·c++·算法
ShiinaMashirol6 小时前
代码随想录打卡|Day27(合并区间、单调递增的数字、监控二叉树)
java·算法
wuqingshun3141598 小时前
蓝桥杯 5. 交换瓶子
数据结构·c++·算法·职场和发展·蓝桥杯
Demons_kirit8 小时前
Leetcode 2845 题解
算法·leetcode·职场和发展
adam_life9 小时前
http://noi.openjudge.cn/——2.5基本算法之搜索——200:Solitaire
算法·宽搜·布局唯一码
我想进大厂9 小时前
图论---朴素Prim(稠密图)
数据结构·c++·算法·图论
我想进大厂9 小时前
图论---Bellman-Ford算法
数据结构·c++·算法·图论
AIGC大时代9 小时前
高效使用DeepSeek对“情境+ 对象 +问题“型课题进行开题!
数据库·人工智能·算法·aigc·智能写作·deepseek
CODE_RabbitV10 小时前
【深度强化学习 DRL 快速实践】近端策略优化 (PPO)
算法
Wendy_robot11 小时前
【滑动窗口+哈希表/数组记录】Leetcode 438. 找到字符串中所有字母异位词
c++·算法·leetcode