http://codeforces.com/gym/101149/problem/G
一开始还以为要用二分图去做,但是复杂度也太高了,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; }