湫湫系列故事——消灭兔子 HDU - 4544(贪心+优先队列)

湫湫系列故事——消灭兔子 HDU - 4544

湫湫减肥
  越减越肥!
  最近,减肥失败的湫湫为发泄心中郁闷,在玩一个消灭免子的游戏。
  游戏规则很简单,用箭杀死免子即可。
  箭是一种消耗品,已知有M种不同类型的箭可以选择,并且每种箭都会对兔子造成伤害,对应的伤害值分别为Di(1 <= i <= M),每种箭需要一定的QQ币购买。
  假设每种箭只能使用一次,每只免子也只能被射一次,请计算要消灭地图上的所有兔子最少需要的QQ币。
Input
输入数据有多组,每组数据有四行;
第一行有两个整数N,M(1 <= N, M <= 100000),分别表示兔子的个数和箭的种类;
第二行有N个正整数,分别表示兔子的血量Bi(1 <= i <= N);
第三行有M个正整数,表示每把箭所能造成的伤害值Di(1 <= i <= M);
第四行有M个正整数,表示每把箭需要花费的QQ币Pi(1 <= i <= M)。

特别说明:
1、当箭的伤害值大于等于兔子的血量时,就能将兔子杀死;
2、血量Bi,箭的伤害值Di,箭的价格Pi,均小于等于100000。
Output
如果不能杀死所有兔子,请输出”No”,否则,请输出最少的QQ币数,每组输出一行。
Sample Input
3 3
1 2 3
2 3 4
1 2 3
3 4
1 2 3
1 2 3 4
1 2 3 1
Sample Output
6
4

思路:第一次看到题我直接懵了,虽然知道要在所有能杀死兔子的箭中找到最便宜的,但是不知道怎么用代码表示,还好有有点队列,我们可以将所有能杀死兔子的箭入队,然后每次出队都出队最便宜的箭。我们可以对兔子的血量进行降序排序,对于箭的伤害进行降序排序。(如果不进行排序的话对于每只兔子我们都需要把所有的箭遍历一遍才能找完所有能杀死这个兔子的箭),排序之后对于某个兔子,如果有一个箭杀不死它,那么这个箭之后的都杀不死它,省去好多时间。

#include<iostream>
#include<algorithm>
#include<queue>
struct node {
	int di, pi;//伤害和价格
}arr[100005];
int brr[100005];
using namespace std;
bool cmp1(node a, node b) {
	return a.di > b.di;
};
bool cmp2(int a, int b) {
	return a > b;
}
int main() {
	int n, m; 
	while (cin >> n >> m) {
		for (int i = 0; i < n; i++) cin >> brr[i];
		for (int i = 0; i < m; i++)cin >> arr[i].di;
		for (int i = 0; i < m; i++)cin >> arr[i].pi;
		sort(arr, arr + m, cmp1);
		sort(brr, brr + n, cmp2);
		long long price = 0; int flag = 1;
		int j = 0;//箭的下标
		priority_queue<int, vector<int>, greater<int> > q;
		for (int i = 0; i < n; i++) {
			while (j < m&&arr[j].di >= brr[i]) {//入队所有可以杀死这个兔子的箭的价格
				q.push(arr[j].pi);
				j++;
			}
			if (q.empty()) {
				flag = 0;
				break;
			}
			price += q.top();
			q.pop();
		}
		if (flag)cout << price << endl;
		else cout << "No" << endl;
	}
	
	return 0;
}

猜你喜欢

转载自blog.csdn.net/m0_47642376/article/details/107779525