HDU 4544|湫湫系列故事——消灭兔子|平衡树|优先队列|STL

湫湫系列故事——消灭兔子

Time Limit: 3000/1000 MS (Java/Others)
Memory Limit: 65535/32768 K (Java/Others)

Problem Description

　　湫湫减肥
　　越减越肥！
　　
　　最近，减肥失败的湫湫为发泄心中郁闷，在玩一个消灭免子的游戏。
　　游戏规则很简单，用箭杀死免子即可。
　　箭是一种消耗品，已知有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

Source

2013腾讯编程马拉松复赛第三场（3月31日）

Solution

数据范围挺大的，可以考虑贪心。
显然我们可以将箭的价格从小到大排序，并用这些箭杀死其能杀死的最大血量的兔子，就可以了。维护兔子可以用multiset。

---

第一种方法是先考虑了箭，我们这次考虑兔子，也是先处理血量最大的兔子，对于每只兔子，我们找出能杀死这个兔子的最便宜的箭杀死这只兔子就可以了，如果没有能杀死的就No。priority_queue就可以维护了。

---

貌似multiset比priority_queue快。。

#include <cstdio>
#include <set>
#include <algorithm>
typedef long long ll;
using namespace std;
#define FOR(i,j,k) for(i=j;i<=k;++i)

struct Arrow {
    int damage, price;
    bool operator <(const Arrow &b) const {
        return price == b.price ? damage > b.damage : price < b.price;
    }
} arrows[100005];

multiset<int> rabbits;

int main() {
    int n, m, i, j;
    long long ans;
    while (scanf("%d%d", &n, &m) == 2) {
        ans = 0; rabbits.clear();
        FOR(i,1,n) scanf("%d", &j), rabbits.insert(j);
        FOR(i,1,m) scanf("%d", &arrows[i].damage);
        FOR(i,1,m) scanf("%d", &arrows[i].price);
        sort(arrows + 1, arrows + m + 1);
        if (m < n) { puts("No"); continue; }
        FOR(i,1,m) {
            if (rabbits.empty()) break;
            multiset<int>::iterator it = rabbits.upper_bound(arrows[i].damage);
            if (it == rabbits.begin()) continue;
            else --it;
            rabbits.erase(it); ans += arrows[i].price;
        }
        if (rabbits.empty())
            printf("%I64d\n", ans);
        else
            puts("No");
    }
    return 0;
}