贪心算法--项目花费与收益问题

思路:将所有项目按照花费从小到大排序,选出所有的可以做的项目,然后找一个收益最大的。

现在资金增多了,继续选出所有的可以做的项目,再找一个收益最大的。直到k等于0;

实现:将输入两个数组中的每个位置的花费和收益两个值结合成一个节点。按照花费从小到大放入小根堆,如果花费小于现有资金就放入一个利益从大到小排序的大根堆中。然后从大根堆中弹出一个最大利益的值。现有资金增加,然后循环直到k等于0;

public static class Node{//项目
		int profit;
		int cost;
		public Node(int profit,int cost) {
			this.profit = profit;
			this.cost = cost;
		}
	}
	//利润比较器
	public static class ProfitComparator implements Comparator<Node>{

		@Override
		public int compare(Node o1, Node o2) {
			// TODO Auto-generated method stub
			return o2.profit - o1.profit;
		}
		
	}
	
	//花费比较器
	public static class CostComparator implements Comparator<Node>{

		@Override
		public int compare(Node o1, Node o2) {
			// TODO Auto-generated method stub
			return o1.cost - o2.cost;
		}
		
	}
	
	
	public static int findMax(int K,int W,int[] profits,int[] costs) {
		PriorityQueue<Node> minCost = new PriorityQueue<Code02_IPO.Node>(new CostComparator());
		PriorityQueue<Node> MaxProfit = new PriorityQueue<Code02_IPO.Node>(new ProfitComparator());
		
		for(int i=0;i<profits.length;i++) {//将所有的项目放到按花费排序的小根堆中
			minCost.add(new Node(profits[i],costs[i]));
		}
		for(int i=0;i<K;i++) {
			//peek表示查看小根堆中的第一个元素。并不弹出。
			while(!minCost.isEmpty()&&minCost.peek().cost<=W) {
				MaxProfit.add(minCost.poll());
			}
			//把花费小的都弹如大根堆中了,但是还是没有能做的项目就直接返回了。
			if(MaxProfit.isEmpty()) {
				return W;
			}
			W += MaxProfit.poll().profit;
			
		}
		
		
		
		return W;
	}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值