1139 First Contact（30分）

题目翻译：

A喜欢B，但是A不会直接找B，A要找他的朋友且同性别的C，然后C去找和B的朋友且同性别的D，让D去联系B，给定 A和B，求C和D有哪些（竟然可以同性恋）

话说这道题没考虑一种情况，如果A和B事先是朋友的话，还需要人帮忙传话么？不懂

题解思路：

先对每个人都将每个人的朋友都存到一个集合中形成一个map，然后对A和B的集合开始遍历。

注意还要另外用一个map（哈希表）记录朋友关系

代码：

#include <bits/stdc++.h>
using namespace std;
unordered_map<string, bool> g; // 0表示女，1是男
unordered_map<string, set<string>> p;
map<pair<string, string>, bool> r;
int N, M, K;

bool comp(pair<string, string> p1, pair<string, string> p2)
{
	if (p1.first != p2.first) return p1.first <= p2.first;
	else return p1.second < p2.second;
}

int main()
{
	scanf("%d %d", &N, &M);
	string s1, s2;
	for (int i = 0; i < M; i++)
	{
		cin >> s1 >> s2;
		if (s1[0] != '-') g[s1] = 1;
		if (s2[0] != '-') g[s2] = 1;
		p[s1].insert(s2);
		p[s2].insert(s1);
		r[make_pair(s1, s2)] = 1;
		r[make_pair(s2, s1)] = 1;
	}
	scanf("%d", &K);
	while (K--)
	{
		vector<pair<string, string>> v;
		cin >> s1 >> s2;
		int g1 = g[s1], g2 = g[s2];
		for (auto& i : p[s1])
		{
			if (i != s2 && g[i] == g1)
				for (auto& j : p[s2])
				{
					if (g[j] == g2 && j != s1 && r[make_pair(i, j)])
					{
						string s1 = i, s2 = j;
						if (s1[0] == '-') s1 = s1.substr(1);
						if (s2[0] == '-') s2 = s2.substr(1);
						v.push_back(make_pair(s1, s2));
					}
				}
		}
		printf("%d\n", v.size());
		sort(v.begin(), v.end(), comp);
		for (auto i : v)
			printf("%s %s\n", i.first.c_str(), i.second.c_str());
	}
	return 0;
}

坑点：

对于测试点3、4：注意C不能B，A不能是D这一个隐含条件。

对于测试点5超时：多剪枝。