PAT A1139 First Contact (30 分) 图

最新推荐文章于 2021-08-16 11:41:50 发布

原创最新推荐文章于 2021-08-16 11:41:50 发布 · 347 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

PAT 同时被 2 个专栏收录

165 篇文章

订阅专栏

Review

60 篇文章

订阅专栏

本文介绍了一种基于图论的交友算法，通过构建邻接表，寻找两人间的共同朋友路径，解决特定交友问题。该算法利用两次遍历找到连接两人的中间人，确保性别和朋友关系的正确性。

题目大意：如果A想认识B，需要通过中间人C和D的联系。其中，C与A同性别，并且是A的朋友。D与B同性别，并且是B的朋友。C和D也是朋友。现在以0~9999的id表示N个人，女性的id前以符号标记。给出一些id对，判断他们能否互相认识。

如果把人看成顶点，题目用图的邻接表很容易得到思路，首先考察A的所有邻接节点，其中与A同性的，再考察他的邻接节点，如果有与B同性的，并且也是B的邻接节点的节点，那么就找到了一对C和D。因此只需要两次遍历就可以找到一对解。但是，有以下几个细节需要注意：

① 输入的id中，如果是-0000，无法通过读入整数来判断性别，因此只能通过读入字符串的形式判断性别。之后，-0000~-9999要怎么映射为非负整数，可以用 10000 + 绝对值的方式，这样，男性的id 就是0-9999,女性的id就是10000-19999。

② C是A的朋友，并且要与A同性，这可以直接通过遍历A的邻接节点加一个判断条件实现。D是C的朋友，并且要与B同性，也是B的朋友，因此在遍历C的邻接节点时要加两个判断条件。

③ 遍历邻接节点时可能产生环，或者直接找到A和B其中一个。由A找到C，要排除C=B的情况；由C找到D，要排除D=A或D=B的情况。因此，判断条件中要加上C不是A或B，D也不是A或B。

④ 输出id是4位整数。

⑤ 用set<int> 存储，可以更简洁的实现判断是否是朋友这个步骤。

AC代码：

#include <bits/stdc++.h>

using namespace std;

const int MAXN = 20010;
unordered_set<int> G[MAXN];

bool isSameGender(int a, int b)
{
    return ((a <= 9999 && b <= 9999) || (a >= 10000 && b >= 10000));
}

void findCD(int a, int b, vector<pair<int, int>> &v)
{
    for(auto ita = G[a].begin(); ita != G[a].end(); ita++)
    {
        int c = *ita;
        if(c != a && c != b && isSameGender(a, c))
        {
            for(auto itc = G[c].begin(); itc != G[c].end(); itc++)
            {
                int d = *itc;
                if(d != a && d != b && d != c && isSameGender(b, d) && G[b].count(d) > 0)
                {
                    if(c >= 10000) c -= 10000;
                    if(d >= 10000) d -= 10000;
                    v.push_back(make_pair(c, d));
                }
            }
        }
    }
}

int main()
{
    int N, M, K;
    scanf("%d%d", &N, &M);
    for (int i = 0; i < M; ++i)
    {
        string id1, id2;
        cin >> id1 >> id2;
        int u, v;
        if(id1[0] == '-') u = 10000 - stoi(id1);
        else u = stoi(id1);
        if(id2[0] == '-') v = 10000 - stoi(id2);
        else v = stoi(id2);
        G[u].insert(v);
        G[v].insert(u);
    }
    scanf("%d", &K);
    for (int i = 0; i < K; ++i)
    {
        string id1, id2;
        cin >> id1 >> id2;
        int u, v;
        if(id1[0] == '-') u = 10000 - stoi(id1);
        else u = stoi(id1);
        if(id2[0] == '-') v = 10000 - stoi(id2);
        else v = stoi(id2);
        vector<pair<int, int>> ans;
        findCD(u, v, ans);
        sort(ans.begin(), ans.end());
        printf("%d\n", ans.size());
        for (int j = 0; j < ans.size(); ++j)
        {
            printf("%04d %04d\n", ans[j].first, ans[j].second);
        }
    }
    return 0;
}