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

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

题目思路：

        对于一个结点，计算他的子树（包含他），他最少排第几名。

        怎么转移呢？？

        对于叶子节点在他的子树中肯定排名第一，所以dp【叶子】 =  1  ；对于上边的x结点，如果是max标记，那么我要取他儿子中排名最小的那个，他可以选择排在最前边，也就是dp【x】=min(dp【x】，dp【x_son】)。如果是min标记，那么我要取儿子中排名之和，因为全部儿子的都应该比这个排名小。

#include<iostream>
#include<string.h>
#include<algorithm>
#include<vector>
#include<stdio.h>
#include<cmath>
#include<queue>
using namespace std;
const int MAXN=3e5+5;
vector<int>v[MAXN];
int num;
int lea[MAXN],val[MAXN],vis[MAXN],dp[MAXN];
void dfs(int x)
{
    int len=v[x].size();
    if(len==1){
        dp[x]=1;
        num++;
        return ;
    }
    lea[x]=1;
    if(val[x])dp[x]=0x7fffffff;
    for(int i=0;i<len;i++){
        int now=v[x][i];
        if(!vis[now]){
            vis[now]=1;
            dfs(now);
            if(val[x])dp[x]=min(dp[x],dp[now]);
            else dp[x]+=dp[now];
        }
    }
}
int main()
{
    int n;
    cin>>n;
    //memset(dp,0x3f,sizeof(dp));
    for(int i=1;i<=n;i++){
        cin>>val[i];
    }
    for(int i=2;i<=n;i++){
        int a;cin>>a;
        v[a].push_back(i);
        v[i].push_back(a);
    }
    vis[1]=1;
    dfs(1);
    cout<<num+1-dp[1]<<endl;
}