SGU 548 Dragons and Princesses(贪心)

坑爹:

误以为王子可以marry中间任意一个princess,觉得不可搞,然后问了Lee,觉得优先队列也不好搞。。。

反思:

问了wintowanti,发现看错题了。。。其实只能marry最后一个princess。。。这样的话,用优先队列贪心就行了,每碰到一个di,将gi加入队列,每碰到一个pi,强制维护队列size < bi就行了。。。嗯嗯。。。

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<queue>
#define REP(i, n) for(int i=0; i<n; i++)
using namespace std;

int n, x;
char op[2];

struct Node
{
    int g, id;
    Node(){}
    Node(int g, int id):g(g), id(id){}
    bool operator < (const Node& rhs) const
    {
        return g > rhs.g;
    }
};
priority_queue<Node> q;

int main()
{
    scanf("%d", &n);
    REP(i, n-2)
    {
        scanf("%s%d", op, &x);
        if(op[0] == 'd') q.push(Node(x, i + 2));
        else while(q.size() >= x) q.pop();
    }
    scanf("%s%d", op, &x);
    if(q.size() < x) puts("-1");
    else
    {
        vector<int> path;
        int ans = 0;
        while(!q.empty())
        {
            ans += q.top().g;
            path.push_back(q.top().id);
            q.pop();
        }
        sort(path.begin(), path.end());
        int nc = path.size();
        printf("%d\n%d\n", ans, nc);
        REP(i, nc) printf("%d%c", path[i], i == nc - 1 ? '\n' : ' ');
    }
    return 0;
}


你可能感兴趣的:(SGU 548 Dragons and Princesses(贪心))