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 = 1;
bool fri[10000][10000];
vector<int>samefri[10000]; 

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 = 1; 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 = 0;
        vector<node>ans;
        int first = abs(stoi(a)), second = abs(stoi(b));
        for(int i = 0; i < samefri[first].size(); i++){
            int c = samefri[first][i];
            if(c == second)continue;
            for(int j = 0; 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 = 0; i < cnt; i++){
            printf("%04d %04d\n", ans[i].c, ans[i].d);
        }
    }
    return 0;
}