PAT A 1139 First Contact (30 分)

一、题目概述

当A想向B告白时，需要通过A同性朋友C，C再通过B的同性朋友D转达给B。
依据给定关系网络，和一对有情人A、B，找到所有可以为他们牵线搭桥的朋友对。

二、思路

用数组related[]标记出B的所有直接同性好友（注意A、同性时不要标记A），遍历A的直接同性好友（同样要排除B），若A的直接同性好友C的朋友D被标记，C、D是一对牵线人。

测试点1答案错误：注意-0000也是女生。

三、代码

#include <cstdio>
#include <vector>
#include <algorithm>
#include <cstdlib>
using namespace std;
int cmp( vector<int> a, vector<int> b )
{
	if( a[0] != b[0] )
		return a[0] < b[0];
	return a[1] < b[1];
}
int main()
{
	int N, M, K;
	scanf("%d %d", &N, &M);
	vector<bool> gender(10000);
	vector<vector<int>> G(10000), ans;
	for( int i = 0, v1, v2; i < M; ++i )
	{
		char str1[6], str2[6];
		scanf("%s %s", str1, str2);
		v1 = abs(atoi(str1)), v2 = abs(atoi(str2));
		gender[v1] = ( str1[0] != '-' );
		gender[v2] = ( str2[0] != '-' );
		G[v1].push_back(v2);
		G[v2].push_back(v1);
	}
	scanf("%d", &K);
	for( int i = 0, v1, v2; i < K; ++i )
	{
		scanf("%d %d", &v1, &v2);
		ans.clear();
		v1 = abs(v1), v2 = abs(v2);
		vector<bool> tag(10000, false);
		for( int j = 0; j < G[v2].size(); ++j )
			if( G[v2][j] != v1 && gender[v2] == gender[G[v2][j]] )
				tag[G[v2][j]] = true;
		for( int j = 0, v; j < G[v1].size(); ++j )
			if( G[v1][j] != v2 && gender[v1] == gender[G[v1][j]] )
			{
				v = G[v1][j];
				for( int k = 0; k < G[v].size(); ++k )
					if( tag[G[v][k]] )
						ans.push_back({v, G[v][k]});
			}
		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][0], ans[j][1]);
	}
}