PAT 1139. First Contact (30) 模拟

传送门

题意：

给n个人，m对朋友关系（有负号的代表女孩）。

k组询问，每组询问给出一对关系（a,b）表示a追求b，追求方式：a在认识的人中找一个和a自己同性的c，c在认识的人中找到一个和b同性并且认识b的d。

简单来说就是a认识c，c认识d，d认识b，并且a和c同性，d和b同性

思路:

很水的一个题目只是模拟即可.我们只需要分别存储下a、b的同性朋友,顺便用map标记一下异性之间是否为朋友,然后暴力找就行了。

唯一的坑点就是:

    两个人的id是可以重号的,如-0000 和0000.只是性别不同,这是如果用int会看成一个人,所以这里需要用字符串读入一下,当长度相等就说明是同性的了,同时用stoi将其转化为整数。

    

#include<bits/stdc++.h>
#define pb push_back
using namespace std;
const int maxn = 1e4+5;

map<int,int>mp[maxn];

vector<int>vt[maxn];
struct gend
{
    int id1,id2;
    gend(int x = 0,int y = 0)
    {
        id1 = x,id2 = y;
    }
};
vector<gend>ans;
int n,m,k;
void init()
{
    for(int i = 0; i< maxn;++i)
    {
        vt[i].clear();
        mp[i].clear();
    }
    return ;
}
bool cmp(gend a,gend b)
{
    if(a.id1 == b.id1)
        return a.id2 < b.id2;
    return a.id1 < b.id1;
}
int main()
{
    while(~scanf("%d %d",&n,&m))
    {
        init();
        for(int i = 0;i < m;++i)
        {
            string  a,b;
            cin >> a >> b;
            int aa,bb;
            aa = abs(stoi(a));
            bb = abs(stoi(b));
            if(a.size() == b.size())
            {
                vt[aa].pb(bb);
                vt[bb].pb(aa);
            }
            mp[aa][bb] = mp[bb][aa] = 1;
        }
        scanf("%d",&k);
        for(int i = 0;i < k;++i)
        {
            ans.clear();
            int a,b;
            scanf("%d %d",&a,&b);
            a = abs(a),b = abs(b);
            for(int j = 0;j < vt[a].size();++j)
            {
                int c = vt[a][j];
                if(c == a || c == b) continue;
                for(int kk = 0;kk < vt[b].size();++kk)
                {
                    int d = vt[b][kk];
                    if(c == d || d == a || d == b) continue;
                    if(mp[c][d] == 1 || mp[d][c] == 1)
                        ans.pb(gend(c,d));
                }
            }
            sort(ans.begin(),ans.end(),cmp);
            printf("%d\n",ans.size());
            for(int j = 0;j < ans.size();++j)
            {
                printf("%04d %04d\n",ans[j].id1,ans[j].id2);
            }
        }
    }
    return 0;
}