L2-016 愿天下有情人都是失散多年的兄妹

最新推荐文章于 2024-09-11 15:01:00 发布

前路无畅

最新推荐文章于 2024-09-11 15:01:00 发布

阅读量103

点赞数

分类专栏： PTA天梯赛往届题目文章标签：数据结构算法 c++

本文链接：https://blog.csdn.net/ygh2437068791/article/details/130177925

版权

PTA天梯赛往届题目专栏收录该内容

3 篇文章 0 订阅

订阅专栏

该程序使用深度优先搜索(DFS)遍历两个人的家族树，检查五代内是否有共同祖先或性别相同的两人。通过构建地图m存储关系，用m1记录性别，对于每个查询，分别从两个人开始进行DFS，如果找到相同节点则返回不能结婚，否则可以结婚。

摘要由CSDN通过智能技术生成

L2-016 愿天下有情人都是失散多年的兄妹

题目大意：给出两个人，判断两个人能否结婚（即五代之内有没有一样的人或者性别是否相同）

做法：每次给出两个人的时候，dfs出他们的祖上都有谁，然后暴力比较有没有相同的人就行，注意性别的保存，输入数据的时候，自己的父母的性别也要标记下，我第一次做的时候就是没有标记，只得了17分，代码如下：

/*
coder:sunshine
school:njupt
*/
#include <bits/stdc++.h>
using namespace std;
#define endl '\n' //交互题删掉
#define x first
#define y second
typedef long long ll;
typedef pair<int, int> pii;
constexpr int mod = 1e9 + 7;

map<int, vector<int>> m;
map<int, char> m1;
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++)
    {
        int a, b, c;
        char d;
        cin >> a >> d >> b >> c;
        m1[a] = d;
        m[a].push_back(b), m1[b] = 'M';
        m[a].push_back(c), m1[c] = 'F';
    }
    int q;
    cin >> q;
    while (q--)
    {
        int a, b;
        cin >> a >> b;
        if (m1[a] == m1[b])
            cout << "Never Mind" << endl;
        else
        {
            vector<int> s1, s2;
            function<void(int, int)> dfs1 = [&](int a, int sum)
            {
                if (sum == 5)
                    return;
                if (a != -1)
                {
                    s1.push_back(a);
                    for (auto c : m[a])
                    {
                        dfs1(c, sum + 1);
                    }
                }
            };
            function<void(int, int)> dfs2 = [&](int a, int sum)
            {
                if (sum == 5)
                    return;
                if (a != -1)
                {
                    s2.push_back(a);
                    for (auto c : m[a])
                    {
                        dfs2(c, sum + 1);
                    }
                }
            };
            dfs1(a, 0);
            dfs2(b, 0);
            bool ok = 1;
            for (auto c : s1)
            {
                for (auto d : s2)
                {
                    if (c == d)
                    {
                        ok = 0;
                    }
                }
            }
            if (ok)
                cout << "Yes" << endl;
            else
                cout << "No" << endl;
        }
    }
    return 0;
}