CodeM美团点评编程大赛初赛B轮 黑白树【DFS深搜+暴力】

[编程题] 黑白树

时间限制：1秒

空间限制：32768K

一棵n个点的有根树，1号点为根，相邻的两个节点之间的距离为1。树上每个节点i对应一个值k[i]。每个点都有一个颜色，初始的时候所有点都是白色的。
你需要通过一系列操作使得最终每个点变成黑色。每次操作需要选择一个节点i，i必须是白色的，然后i到根的链上（包括节点i与根）所有与节点i距离小于k[i]的点都会变黑，已经是黑的点保持为黑。问最少使用几次操作能把整棵树变黑。

输入描述:

第一行一个整数n (1 ≤ n ≤ 10^5)
接下来n-1行，每行一个整数，依次为2号点到n号点父亲的编号。
最后一行n个整数为k[i] (1 ≤ k[i] ≤ 10^5)

样例解释:
对节点3操作，导致节点2与节点3变黑
对节点4操作，导致节点4变黑
对节点1操作，导致节点1变黑

输出描述:

一个数表示最少操作次数

输入例子:

4
1
2
1
1 2 2 1

输出例子:

3

题目链接：https://www.nowcoder.com/question/next?pid=5599304&qid=105269&tid=8919747

分析：DFS深搜+暴力写法，可以参考一下的啦！

其实就是一个DFS递归啊，

子树全变黑的最优解：sum1，

在最优解sum1下最多还能向上延伸的点数，sum2，

sum3是，子树下不被选择的点作为备选的点，最多还能向上延伸的点数！

下面给出AC代码：

#include<bits/stdc++.h>
using namespace std;
const int N =166666;
typedef long long ll;
vector<ll> T[N];
ll n,a[N];
struct Tree
{
    ll ans;
    ll maxn;
    ll Gmaxn;
};
inline ll read()
{
    ll x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
inline void write(ll x)
{
    if(x<0)
    {
        putchar('-');
        x=-x;
    }
    if(x>9)
    {
        write(x/10);
    }
    putchar(x%10+'0');
}
Tree DFS(ll xx,ll yy)
{
    Tree ret={0,0,0};
    ll sum1=0,sum2=-1,sum3=-1;
    for(ll i=0;i<T[xx].size();i++)
    {
        ll to=T[xx][i];
        if(to==yy)
            continue;
        ret=DFS(to,xx);
        sum1+=ret.ans;
        sum2=max(ret.maxn,sum2);
        sum3=max(ret.Gmaxn,sum3);
    }
    if(sum2<=-1)
    {
        sum2=max(a[xx],sum3);
        sum1+=1;
    }
    sum3=max(sum3,a[xx]);
    ret.ans=sum1;
    ret.maxn=sum2-1;
    ret.Gmaxn=sum3-1;
    return ret;
}
int main()
{
    ll p;
    n=read();
    for(ll i=1;i<=n-1;i++)
    {
        p=read();
        T[p].push_back(i+1);
    }
    for(ll i=1;i<=n;i++)
    {
        a[i]=read();
        a[i]--;
    }
    write(DFS(1,0).ans);
    return 0;
}