PAT甲级1139 First Contact

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

题意：

有m对朋友关系，每个人用4为数字的编号表示，如果是负数表示这是女生。

给定k个查询，对于要查的人a和b，问有多少对朋友（c,d）使得c和a是同性，d和b是同性，且c和d是朋友。

思路：

枚举a的所有同性朋友，枚举b的所有同性朋友，看他们是不是朋友。

首先用了map离散化了一下给了个id，可能是这里搞来搞去T了最后一组数据。

实际上简单点可以直接用c++ 11标准里的stoi来做，感觉PAT老是碰到这个函数。

由于标号是唯一的，所以一个人只需要存储他的同性朋友就可以了。

开bool的话二维的1e5也是开的下的。

要注意枚举的时候如果直接碰到了a或b都要跳过。

 #include<cstdio>
 #include<cstdlib>
 #include<map>
 #include<set>
 #include<iostream>
 #include<cstring>
 #include<algorithm>
 #include<vector>
 #include<cmath>
 #include<stack>
 #include<queue>

 #define inf 0x7fffffff
 using namespace std;
 typedef long long LL;
 typedef pair<string, string> pr;

 int n, m, k;
 int id = ;
 bool fri[][];
 vector<int>samefri[]; 

 struct node{
     int c, d;
     node(){
     }
     node(int _c, int _d){
         c = _c;
         d = _d;
     }
 };

 bool cmp(node a, node b)
 {
     if(a.c == b.c)return a.d < b.d;
     else return a.c < b.c;
 }

 int main()
 {
     scanf("%d%d", &n, &m);
     for(int i = ; i <= m; i++){
         string a, b;
         cin>>a>>b;
         int first = abs(stoi(a));
         int second = abs(stoi(b));

         fri[first][second] = true;
         fri[second][first] = true;
         if(a.length() == b.length()){
             samefri[first].push_back(second);
             samefri[second].push_back(first);
         }
     }

     scanf("%d", &k);
     while(k--){
         string a, b;
         cin>>a>>b;

         int cnt = ;
         vector<node>ans;
         int first = abs(stoi(a)), second = abs(stoi(b));
         for(int i = ; i < samefri[first].size(); i++){
             int c = samefri[first][i];
             if(c == second)continue;
             for(int j = ; j < samefri[second].size(); j++){
                 int d = samefri[second][j];
                 if(d == first)continue;
                 if(fri[c][d]){
                     cnt++;
                     ans.push_back(node(c, d));
                     //cout<<a<<" "<<b<<endl<<endl;
                 }
             }
         }

         printf("%d\n", cnt);
         sort(ans.begin(), ans.end(), cmp);
         for(int i = ; i < cnt; i++){
             printf("%04d %04d\n", ans[i].c, ans[i].d);
         }
     }
     return ;
 }