贪心算法-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;
   }
}
相关推荐
算AI14 小时前
人工智能+牙科:临床应用中的几个问题
人工智能·算法
hyshhhh16 小时前
【算法岗面试题】深度学习中如何防止过拟合?
网络·人工智能·深度学习·神经网络·算法·计算机视觉
杉之17 小时前
选择排序笔记
java·算法·排序算法
烂蜻蜓17 小时前
C 语言中的递归:概念、应用与实例解析
c语言·数据结构·算法
OYangxf17 小时前
图论----拓扑排序
算法·图论
我要昵称干什么17 小时前
基于S函数的simulink仿真
人工智能·算法
AndrewHZ18 小时前
【图像处理基石】什么是tone mapping?
图像处理·人工智能·算法·计算机视觉·hdr
念九_ysl18 小时前
基数排序算法解析与TypeScript实现
前端·算法·typescript·排序算法
守正出琦18 小时前
日期类的实现
数据结构·c++·算法
ChoSeitaku18 小时前
NO.63十六届蓝桥杯备战|基础算法-⼆分答案|木材加工|砍树|跳石头(C++)
c++·算法·蓝桥杯