获得的最大钱数

参数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;
		
	}

猜你喜欢

转载自blog.csdn.net/qq_42403295/article/details/88959154