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;
}