2022河南省赛 J. Mex Tree

题意：给你一颗树，树上点权是0~n-1的一个排列，对于k∈[0,n]，让你求出最大的非空联通子图S使得mex(S)=k。

分析：对于mex(S)=k的S，首先不能包含点权为k的节点u，且权值∈[0,k-1]的节点必须全都在以u为根的一棵子树当中，考虑换根。对于根节点u，记其每棵子树的最小值dp1[v]，如果dp1[v]<=a[u]-1，代表以v为根的子树是当前的选择之一，当且仅当只有一种选择的时候，|S|=sz[v]。特别的，当k=0,n时需要特判。换根的同时维护每个子树的最小值，只需再维护次小值即可。

代码：

vector<int>e[maxn];
int a[maxn];
int sz[maxn];
int dp1[maxn],dp2[maxn];
int ans[maxn];
int n;

void update(int u,int val)
{
    if(val<dp1[u])
    {
        dp2[u]=dp1[u];
        dp1[u]=val;
    }
    else if(val<dp2[u])
    {
        dp2[u]=val;
    }
}

void dfs(int u,int fa)
{
    sz[u]=1;
    dp1[u]=a[u],dp2[u]=9e18;
    for(auto v:e[u])
    {
        if(v==fa) continue;
        dfs(v,u);
        update(u,dp1[v]);
        sz[u]+=sz[v];
    }
}

void dfs2(int u,int fa)
{
    int cnt=0,p=0;
    for(auto v:e[u])
    {
        if(a[u]==0) ans[0]=max(ans[0],sz[v]);
        if(dp1[v]<=a[u]-1)
        {
            cnt++;
            p=v;
        }
    }
    if(cnt==1) ans[a[u]]=sz[p];

    int pp=0;
    for(auto v:e[u])
    {
        if(v==fa) continue;
        if(dp1[v]!=dp1[u])
        {
            sz[u]=n-sz[v];
            sz[v]=n;
            update(v,dp1[u]);
            dfs2(v,u);
        }
        else pp=v;
    }
    if(pp)
    {
        sz[u]=n-sz[pp];
        sz[pp]=n;
        dp1[u]=dp2[u];
        update(pp,dp1[u]);
        dfs2(pp,u);
    }
}

void solve()
{
    cin>>n;
    for(int i=1;i<=n;i++) cin>>a[i];
    for(int i=2;i<=n;i++)
    {
        int x;
        cin>>x;
        e[x].push_back(i);
        e[i].push_back(x);
    }
    dfs(1,0);
    dfs2(1,0);
    for(int i=0;i<=n-1;i++) cout<<ans[i]<<" ";cout<<n<<endl;
}