参数1正数数组costs, 参数2正数数组profits ,参数3正数k, 参数4正数W
costs[i]表示i号项目的花费 profits[i]表示i号项目在扣除花
费之后还能挣到的钱(利润) k表示你不能并行、只能串行的最多
做k个项目 W表示你初始的资金
说明:你每做完一个项目,马上获得的收益,可以支持你去做下
一个 项目。
输出: 你最后获得的最大钱数。
通过一个小根堆和一个大根堆来实现,首先把所有项目根据cost花费放进小根堆,再判断小根堆的里是否有花费小于启动资金的项目,有则放进大根堆(根据纯利润的大根堆),这样每次从大根堆里取出一个项目完成,获得利润加到W启动资金里,再去小根堆里判断是否有满足条件的项目进入大根堆,直到k为0或者大根堆为空,则没办法执行项目,返回结果。
public static class Node{
public int profit;
public int cost;
public Node(int p,int c) {
profit=p;
cost=c;
}
}
public static class minCostHeapComparator implements Comparator<Node>{
@Override
public int compare(Node o1, Node o2) {
return o1.cost-o2.cost;
}
}
public static class maxProfitHeapComparator implements Comparator<Node>{
@Override
public int compare(Node o1, Node o2) {
return o2.profit-o2.profit;
}
}
//k个项目,w为启动资金
public static int findMaximizedCapital(int k, int W, int[] Profits, int[] Cost) {
Node[] nodes=new Node[Profits.length];
for(int i=0;i<nodes.length;i++)
nodes[i]=new Node(Profits[i], Cost[i]);
PriorityQueue<Node> minCostHeap=new PriorityQueue<>(new minCostHeapComparator());
PriorityQueue<Node> maxProfitHeap=new PriorityQueue<>(new maxProfitHeapComparator());
for(int i=0;i<nodes.length;i++) {
minCostHeap.add(nodes[i]);
}
while(k>0) {
while(!minCostHeap.isEmpty()&&minCostHeap.peek().cost<=W) {
maxProfitHeap.add(minCostHeap.poll());
}
if(maxProfitHeap.isEmpty())
return W;
W+=maxProfitHeap.poll().profit;
k--;
}
return W;
}