团体程序设计天梯赛-练习集-L2-016. 愿天下有情人都是失散多年的兄妹(dfs)

记录一个菜逼的成长。。

题目链接

一看题目，我以为可以用LCA算法写，写了半天，后来发现不是树(woc…)
再看题目，不超过5代，那么直接暴搜咯、、

查询时，标记一个人的不超过5代的祖先，再从另一个人向上搜索在5代内是否有标记过的祖先。

#include <bits/stdc++.h>
using namespace std;
#define ALL(v) (v).begin(),(v).end()
#define cl(a,b) memset(a,b,sizeof(a))
#define clr clear()
#define pb push_back
#define mp make_pair
#define fi first
#define se second
typedef pair<int,int> PII;
const int maxn = 100000 + 10;
int sex[maxn],vis[maxn],used[maxn];
vector<int>tree[maxn];
void dfs1(int x,int cnt)
{
    vis[x] = 1;
    if(cnt == 4){
        return ;
    }
    for( int i = 0; i < tree[x].size(); i++ ){
        if(!vis[tree[x][i]]){
            dfs1(tree[x][i],cnt+1);
        }
    }
}
bool dfs2(int x,int cnt)
{
    if(cnt == 5){
        return false;
    }
    if(vis[x]){
        return true;
    }
    used[x] = 1;
    bool ret = false;
    for( int i = 0; i < tree[x].size(); i++ ){
        if(!used[tree[x][i]]){
            ret = (ret || dfs2(tree[x][i],cnt+1));
        }
    }
    return ret;
}
int main()
{
    int n;
    while(~scanf("%d",&n)){
        int x,y,z;
        char s[10];
        cl(sex,-1);
        for( int i = 0; i < n; i++ ){
            scanf("%d%s%d%d",&x,s,&y,&z);
            sex[x] = (s[0] == 'M' ? 1 : 0);
            if(y != -1){
                tree[x].pb(y);
                sex[y] = 1;
            }
            if(z != -1){
                tree[x].pb(z);
                sex[z] = 0;
            }
        }
        int q;scanf("%d",&q);
        while(q--){
            scanf("%d%d",&x,&y);
            if(sex[x] == sex[y])puts("Never Mind");
            else {
                cl(vis,0);cl(used,0);
                dfs1(x,0);
                bool ans = dfs2(y,0);
                puts(ans?"No":"Yes");
            }
        }
    }
    return 0;
}