PAT——1139 First Contact 甲级

题目

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

题意

A喜欢B，但A不想自己说，所以要找到自己的同性朋友C去求B的同性朋友D（C和D要求是朋友），让D把A的心意说给B听

题目解析

将给出的朋友关系放入vis中（两个四位数字合并为八位数字，借此来表示朋友关系），并将每个人的朋友（取绝对值）放入对应的向量中

判断时，遍历两人的朋友向量（即寻找C和D），通过vis判断C和D是否是朋友，如果是，放入ans向量中。遍历完成后，按照要求对ans排序并输出

注意

相关注意事项可以到参考的文章中查看，博主说的已经很详细了

AC代码

#include<bits/stdc++.h>
using namespace std;
struct node{
	int t1,t2;
};
bool cmp(node x,node y){
	return x.t1==y.t1?x.t2<y.t2:x.t1<y.t1;
}
int main()
{
	int n,m,k;
	scanf("%d %d",&n,&m);
	unordered_map<int,vector<int>> f;
	unordered_map<int,int> vis;
	while(m--)
	{
		string x,y;
		cin>>x>>y;
		if(x.size()==y.size())
		{
			f[abs(stoi(x))].push_back(abs(stoi(y)));
			f[abs(stoi(y))].push_back(abs(stoi(x)));
		}	
		vis[abs(stoi(x))*10000+abs(stoi(y))]=vis[abs(stoi(y))*10000+abs(stoi(x))]=1;	
	}
	scanf("%d",&k);
	while(k--)
	{
		int a,b;
		scanf("%d %d",&a,&b);
		vector<node> ans;
		for(int i=0;i<f[abs(a)].size();i++)
		for(int j=0;j<f[abs(b)].size();j++)
		{
			if(f[abs(a)][i]==abs(b) || f[abs(b)][j]==abs(a)) continue;
			if(vis[ f[abs(a)][i]*10000+f[abs(b)][j] ]) 
				ans.push_back({f[abs(a)][i],f[abs(b)][j]});
		}
		sort(ans.begin(),ans.end(),cmp);
		printf("%d\n",ans.size());
		for(int i=0;i<ans.size();i++)
			printf("%04d %04d\n",ans[i].t1,ans[i].t2);
	}
}

参考（推荐）

PAT 1139. First Contact (30)-PAT甲级真题