PAT A1139 搞基吗？同学

题目：https://pintia.cn/problem-sets/994805342720868352/problems/994805344776077312

这道题，我想说：

你还要我怎样 要怎样

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

我写完这一道题 就关了机

写完你 就很气

我不要你怎样 没怎样

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

因为那是我 不接收的情况

首先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;
	else
		return 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(), ans.end(), cmp);
		printf("%d\n", ans.size());
		for(int j = 0; j < ans.size(); j++){
			printf("%04d %04d\n", ans[j].a, ans[j].b);
		}
	}
	return 0;
}