HDU3290 The magic apple tree【DFS】

题目链接:HDU3290 The magic apple tree

题意:给你一个树,起始叶子结点的值为他的编号本身,开始更新后,如果一个非叶子结点有K个子节点有值,那么他的值更新为这些值中第(k+1)/2小,求根节点的最后值;

分析:从根节点向下DFS,到叶子结点之后向上回溯,裸题;

#include<bits/stdc++.h>
#define pb push_back
using namespace std;
const int maxn=2e4+7;
vector<int> g[maxn];
int n,rt;
int ind[maxn],oud[maxn],num[maxn];
void dfs(int x)
{
    for(auto i:g[x]) dfs(i);
    int k=0,tmp[maxn];
    for(auto i:g[x])
        if(num[i]) tmp[++k]=num[i];
    if(!k) {num[x]=x;return;}
    sort(tmp+1,tmp+1+k);
    num[x]=tmp[(1+k)>>1];
}
void rua()
{
    for(int i=1;i<=n;i++) ind[i]=oud[i]=num[i]=0,g[i].clear();
    for(int i=1;i<=n;i++)
    {
        int x,y;scanf("%d",&x);
        oud[i]=x;
        while(x--) scanf("%d",&y),g[i].pb(y),ind[y]++;
    }
    for(int i=1;i<=n;i++) if(!oud[i]) num[i]=i;
    for(int i=1;i<=n;i++)
        if(!ind[i]) {rt=i;dfs(i);break;}
    printf("%d\n",num[rt]);
}
int main()
{
    while(~scanf("%d",&n)) rua();
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_43813163/article/details/102807933