无语类型,要说有什么亮点,可能就二维存变一维存吧。
题意:A追B,在A的朋友里找一个C,在B的朋友里找一个D,如果C,D是朋友就算一个答案要输出。(二重循环暴力做,所以说无语)(朋友只能为同性别。男生朋友一定为男性)
注意点:注意如果int存id的话,1234 -0000,这一对会被误认为是朋友,所以要用string存,a.size()==b.size()才是朋友。
#include<bits/stdc++.h>
using namespace std;
int n,m,qury;
vector<int> frien[10000];
unordered_map<int,int>mm;
struct node{
int a,b;
};
bool cmp(node x,node y){
return x.a==y.a?x.b<y.b:x.a<y.a;
}
int main(){
cin>>n>>m;
for(int i=0;i<m;i++){
string a,b;
cin>>a>>b;
if(a.size()==b.size()){
frien[abs(stoi(a))].push_back(abs(stoi(b)));
frien[abs(stoi(b))].push_back(abs(stoi(a)));
}
mm[abs(stoi(a))*10000+abs(stoi(b))]=mm[abs(stoi(b))*10000+abs(stoi(a))]=1;
}
cin>>qury;
for(int i=0;i<qury;i++){
int a,b;
cin>>a>>b;
vector<node> vv;
for(auto j:frien[abs(a)]){
for(auto k: frien[abs(b)]){
if(k==abs(a)||abs(b)==j) continue;
if(mm[k*10000+j]){
vv.push_back({j,k});
}
}
}
sort(vv.begin(),vv.end(),cmp);
printf("%d\n",vv.size());
for(auto j:vv){
printf("%04d %04d\n",j.a,j.b);
}
}
return 0;
}