PAT A1139 搞基吗？同学

PAT A1139 搞基吗？同学

题目：

这道题，我想说：

你还要我怎样 要怎样

你千万不要问我搞基的情况

我写完这一道题 就关了机

写完你 就很气

我不要你怎样 没怎样

我希望考试的时候别再这样

因为那是我 不接收的情况

首先0000与-0000是表示两个人，OK，这种坑我能接收，理论上是成立的。但是，如果A,B同性的时候，A要追B，要避免A在同性朋友里面找B的情况，我去，你俩认识，你找人搭个P的桥啊！果然我理解不了同性恋（声明：我尊重同性恋，这是个人的选择，此处无冒犯的意思）。

我忽略了有男女的编号一样的情况。

#include <cstdio>
#include <iostream>
#include <string>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;
const int maxn = 20000;
int n, m, k;
set<int> same[maxn], dif[maxn];
struct p{int a, b;p (int _a, int _b) : a(_a), b(_b){};
};
vector<p> ans;
bool cmp(p a, p b){if(a.a != b.a)return a.a < b.a;elsereturn a.b < b.b;
}
int main(){scanf("%d%d", &n, &m);for(int i = 0; i < m; i++){string a, b;cin>>a>>b;int A, B;sscanf(a.c_str(), "%d", &A);sscanf(b.c_str(), "%d", &B);if(a.size() == 5){A = -A;A += 10000;}if(b.size() == 5){B = -B;B += 10000;}if(a.size() == b.size()){same[A].insert(B);same[B].insert(A);}else{dif[A].insert(B);dif[B].insert(A);}}scanf("%d", &k);for(int i = 0; i < k; i++){string a, b;cin>>a>>b;int A, B;sscanf(a.c_str(), "%d", &A);sscanf(b.c_str(), "%d", &B);if(a.size() == 5){A = -A;A += 10000;}if(b.size() == 5){B = -B;B += 10000;}ans.clear();if(a.size() == b.size()){for(set<int>::iterator it = same[A].begin(); it != same[A].end(); it++){if(*it == B)continue;for(set<int>::iterator it2 = same[B].begin(); it2 != same[B].end(); it2++){if(*it2 == A)continue;if(same[*it].find(*it2) != same[*it].end()){ans.push_back(p(abs(*it%10000), abs(*it2%10000)));}}}}else{for(set<int>::iterator it = same[A].begin(); it != same[A].end(); it++){for(set<int>::iterator it2 = same[B].begin(); it2 != same[B].end(); it2++){if(dif[*it].find(*it2) != dif[*it].end()){ans.push_back(p(abs(*it%10000), abs(*it2%10000)));}}}}sort(ans.begin(), d(), cmp);printf("%dn", ans.size());for(int j = 0; j < ans.size(); j++){printf("%04d %04dn", ans[j].a, ans[j].b);}}return 0;
}