重返天梯-L2-016 愿天下有情人 (25 分)(bfs)

题目描述

原题链接
给出一对情侣的编号，搜索他们是否有共同的祖先，如有则不可通婚

C++

#include <iostream>
#include <queue>
#include <set>
#include <cstring>
using namespace std;
typedef pair<int, int> PII;
const int N = 1e5 + 10;
struct Person{
    int fid, mid, sex;
} user[N];
int n, m;
bool exist[N];

int main() {
    scanf("%d", &n);
    
    for (int i = 0; i < n; i++) {
        int id, sex, fid, mid;
        char c;
        scanf("%d %c %d %d", &id, &c, &fid, &mid);
        if (c == 'M') sex = 0; // 男
        else sex = 1; // 女
        user[id] = {fid, mid, sex};
        exist[id] = true;
        user[fid].sex = 0; // 同时设置父母的性别 
        user[mid].sex = 1;
    }
    
    scanf("%d", &m);
    while(m--){
        int a, b;
        scanf("%d %d", &a, &b);
        
        if (user[a].sex == user[b].sex) {
            cout << "Never Mind" << endl;
            continue;
        }
        
        int flag = 1;
        queue<PII> q; // 维护id和层次
        q.push({a, 1}), q.push({b, 1});
        set<int> s; // 巧用set判断是否正常通婚
        
        while(q.size()) {
            auto t = q.front();
            q.pop();
            int sz = s.size();
            int ver = t.first, level = t.second;
            s.insert(ver);
            if (sz == s.size()) { // 判断是否有共同祖先
                cout << "No" << endl;
                flag = 0;
                break;
            }
            if (!exist[ver]) continue;
            if (level < 5) {
                int f = user[ver].fid, m = user[ver].mid;
                if (f != -1) q.push({f, level+1});
                if (m != -1) q.push({m, level+1});
            }
            
        }
        if (flag) cout << "Yes" << endl;
    }
    
    return 0;
}