Codeforces Round #551 (Div. 2) Serval and Rooted Tree(树形DP)

题目链接：https://codeforces.com/contest/1153/problem/D

 

题目大意：给出n个点，说明每个点的属性，1就是取儿子中最大的值，0就是取最小的值，分配给每个叶子节点1~k（假设有k个叶子，每个叶子的权值不同），问1的值最大是多少

 

题目思路：顺着想想不出来，可以换个思路，用dp[k]存是第k大，如果是叶子那就是第1大，到了上面以后，如果是max，那么就是取儿子中最小的也就是排名最靠前的，如果是min，那么就只能取儿子的排名之和，最后dp[1]就是根节点最靠前的排名，答案就是叶子节点的个数-dp[1]+1

 

以下是代码：

#include <bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define per(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#define inf 0x3f3f3f3f
const int MAXN = 3e5+5;
int n,a[MAXN],x,dp[MAXN],num;
vector<int>v[MAXN];
void dfs(int u,int fa){
    int len=v[u].size(),minn=inf,sum=0;
    rep(i,0,len-1){
        int to=v[u][i];
        if(to==fa)continue;
        dfs(to,u);
        minn=min(minn,dp[to]);
        sum+=dp[to];
    }
    if(minn==inf)dp[u]=1,num++;
    else if(a[u]){
        dp[u]=minn;
    }
    else{
        dp[u]=sum;
    }
}
int main(){
    while(~scanf("%d",&n)){
        memset(dp,0,sizeof(dp));
        num=0;
        rep(i,1,n){
            scanf("%d",&a[i]);
            v[i].clear();
        }
        rep(i,2,n){
            scanf("%d",&x);
            v[i].push_back(x);
            v[x].push_back(i);
        }
        dfs(1,-1);
        printf("%d\n",num-dp[1]+1);
    }
    return 0;
}