G. Of Zorcs and Axes----贪心+STL

来源:互联网 发布:神经内科收入高知乎 编辑:程序博客网 时间:2024/06/05 14:12

转载地址:http://www.cnblogs.com/liuweimingcprogram/p/6044994.html?utm_source=itdadao&utm_medium=referral


大牛的博客,把set用到出神入化


以下来自大牛:

一开始还以为要用二分图去做,但是复杂度也太高了,O(n * m)的话直接爆炸。

考虑贪心,考虑第i个东西优先选一个能选的,而且这个东西的值尽量小。

就是如果要<3, 3>的话,我希望找到有序对<x, y>,其中x和y都最接近3, 3

那么这里就涉及到同时排序两个数,就是这两个数的优先级一样,而且二分找的时候也要同时比较两个数了

一开始还想用set的,但是这样重载运算符的set我没用过,重载成 return a < rhs.a && b < rhs.b; 这样会把有些set的元素载入不了,不知道为何,希望有大牛指点一二。

那么可以这样去想,先对x排序,如果x相同再对y排序,全部相同就按照被选的那个优先,就是m那堆数字有先。

这样做有什么好处呢?可以维护一个set,每次如果是m那堆数字的话,就扔去set那里,如果不是,就从set里面找,找的时候只需要比较y就行了,因为x已经保证严格成立了。

最后说一下set那里,自己写一个比较函数的话,就要考虑重复的元素,因为重复的元素会被自动删除的,但是我本来结构体那里就有了个id这个值,id是保证不同的了,id是我用来记录答案的,但是它还是去重了。现在试了发现,如果想id不同,就当成不同的元素的话,就要在你的cmp那里把id加上去比较,

这里坑了我太久了,一直wa9、

给个数据吧

6
2 5
4 3
4 3
4 3
4 3
4 2
6
5 6
4 3
4 3
4 2
4 3
4 3


代码:

#include <cstdio>#include <cstdlib>#include <cstring>#include <cmath>#include <algorithm>#define IOS ios::sync_with_stdio(false)using namespace std;#define inf (0x3f3f3f3f)typedef long long int LL;#include <iostream>#include <sstream>#include <vector>#include <set>#include <map>#include <queue>#include <string>const int maxn = 5e5 + 20;struct node {    int a, b, id;    int flag;    node() {}    node(int aa, int bb, int iidd, int Orz_zk) : a(aa), b(bb), id(iidd), flag(Orz_zk) {}    bool operator < (const struct node & rhs) const {        if (a != rhs.a) return a < rhs.a;        else if (b != rhs.b) return b < rhs.b;        else return flag < rhs.flag;    }}arr[maxn];struct cmp {    bool operator() (struct node a, struct node b) {        if (a.b != b.b) {            return a.b < b.b;        } else if (a.a != b.a) return a.a < b.a;        else if (a.flag != b.flag) return a.flag < b.flag; //找的时候flag是0,也要处理        else return a.id < b.id; //注意,这个一定要有    }};set<struct node, cmp>ss;int ans[maxn];void work() {    int n;    cin >> n;    for (int i = 1; i <= n; ++i) {        cin >> arr[i].a >> arr[i].b;        arr[i].id = i;        arr[i].flag = 0;    }    int m;    cin >> m;    for (int i = 1; i <= m; ++i) {        cin >> arr[i + n].a >> arr[i + n].b;        arr[i + n].flag = 1;        arr[i + n].id = i;    }    sort(arr + 1, arr + 1 + n + m);//    for (int i = m + n; i >= 1; --i) {//        cout << arr[i].a << " " << arr[i].b << " " << arr[i].flag << " " << arr[i].id << endl;//    }//    cout << endl;    set<struct node> :: iterator it;    for (int i = m + n; i >= 1; --i) {//        for (it = ss.begin(); it != ss.end(); ++it) {//            cout << it->a << " " << it->b << " " << it->id << endl;//        }//        cout << endl;//        cout << arr[i].a << " " << arr[i].b << " " << arr[i].flag << endl;        if (arr[i].flag) {            ss.insert(arr[i]);            continue;        }        if (ss.size() == 0) {            cout << -1 << endl;            return;        }        it = ss.lower_bound(arr[i]);        if (it == ss.end()) {            cout << -1 << endl;            return;        }        ans[arr[i].id] = it->id;        ss.erase(it);    }    for (int i = 1; i <= n; ++i) {        if (ans[i] == 0) while(1);        cout << ans[i] << " ";    }}int main() {#ifdef local    freopen("data.txt","r",stdin);#endif    IOS;    work();    return 0;}



0 0
原创粉丝点击