First Contact

最新推荐文章于 2021-02-07 18:47:04 发布

原创最新推荐文章于 2021-02-07 18:47:04 发布 · 689 阅读

0 ·

CC 4.0 BY-SA版权

PAT 甲级专栏收录该内容

31 篇文章

订阅专栏

本文介绍了一种用于社交网络中查询共同朋友关系的算法实现，通过图论中的邻接表和哈希表来高效地找出两个用户之间的共同朋友，并按一定规则排序输出。

#include<iostream>
#include<vector>
#include<unordered_map>
#include<cmath> 
#include<algorithm>
using namespace std;
vector<int> v[10000];
unordered_map<int,bool> arr;//用数组表示的话数组太稀疏了 
struct node{
	int fa,fb;
};
bool cmp(node x,node y){
	if(x.fa!=y.fa)
		return x.fa<y.fa;
	return x.fb<y.fb;
} 
int main(){
	int N,M,K,p1,p2;
	string a,b;
	scanf("%d%d",&N,&M);
	for(int i=0;i<M;i++){
		cin>>a>>b;
		p1=abs(stoi(a));p2=abs(stoi(b));
		if(a.length()==b.length()){//是同性朋友
			v[p1].push_back(p2);
			v[p2].push_back(p1);
		}
		arr[p1*10000+p2]=arr[p2*10000+p1]=true;
	}
	scanf("%d",&K);
	for(int i=0;i<K;i++){
	  vector<node> ans;
		scanf("%d%d",&p1,&p2);
		p1=abs(p1);p2=abs(p2);
		for(int j=0;j<v[p1].size();j++){
			for(int k=0;k<v[p2].size();k++){
			  int f1=v[p1][j],f2=v[p2][k];
				if(f1==p2||f2==p1) continue;
				if(arr[f1*10000+f2]==true)
					ans.push_back(node{f1,f2});
			}
		}
		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].fa,ans[i].fb);
	}
	return 0;
}