收益最多IPO

版权声明:有一些内容粘贴之后变成图片如果需要原文原视频 可以联系我 https://blog.csdn.net/wodemale/article/details/89712694

输入:
参数1,正数数组 costs
参数2,正数数组profits
参数3,正数k
参数4,正数m

costs[i]表示i号项目的花费
profits[i]表示i号项目在扣除花费之后还能挣到的钱(利润)
k表示你不能并行、只能串行的最多做k个项目
m表示你初始的资金

说明:你每做完一个项目,马上获得的收益,可以支持你去做下一个 项目。

输出: 你最后获得的最大钱数。

public class IPO {
	public static class Node {
		public int p;
		public int c;

		public Node(int p, int c) {
			this.p = p;
			this.c = c;
		}
	}

	public static class MinCostComparator implements Comparator<Node> {
		//花费比较器
		@Override
		public int compare(Node o1, Node o2) {
			return o1.c - o2.c;
		}

	}

	public static class MaxProfitComparator implements Comparator<Node> {
		//利润比较器
		@Override
		public int compare(Node o1, Node o2) {
			return o2.p - o1.p;
		}

	}

	public static int findMaximizedCapital(int k, int W, int[] Profits, int[] Capital) {
		Node[] nodes = new Node[Profits.length];
		for (int i = 0; i < Profits.length; i++) {
			//每个项目一个节点
			nodes[i] = new Node(Profits[i], Capital[i]);
		}
		//小根堆,选择花费小的
		PriorityQueue<Node> minCostQ = new PriorityQueue<>(new MinCostComparator());
		
		//大根堆,选择选择利润大的
		PriorityQueue<Node> maxProfitQ = new PriorityQueue<>(new MaxProfitComparator());
		for (int i = 0; i < nodes.length; i++) {
			minCostQ.add(nodes[i]);
		}
		
		for (int i = 0; i < k; i++) {
			//能做的项目放到大根堆,解锁项目
			while (!minCostQ.isEmpty() && minCostQ.peek().c <= W) {
				maxProfitQ.add(minCostQ.poll());
			}
			
			//项目不一定能做完,大根堆里面没有可以选择做的项目
			//初始资金不够则项目要求停放,退出循环此时i<K,不一定为K
			if (maxProfitQ.isEmpty()) {
				return W;
			}
			//利润增加重新解锁
			W += maxProfitQ.poll().p;
		}
		return W;
	}

}

猜你喜欢

转载自blog.csdn.net/wodemale/article/details/89712694
IPO