链接:https://ac.nowcoder.com/acm/problem/13249
来源:牛客网
题目描述
一棵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变黑
输出描述:
一个数表示最少操作次数
示例1
输入
复制
4
1
2
1
1 2 2 1
输出
复制
3
我们从每个叶子节点往上回溯。
如果子树里面的点染不到这个点就ans++,然后将这个点的染色值传上去。
如果可以染到这个点,就更新父节点的染色值,然后记录当前节点染色到的最大值,并且返回。
代码如下:
#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int maxx=1e5+100;
struct node{
int to;
int next;
}e[maxx<<1];
int tot,head[maxx<<1],k[maxx],fa[maxx],n;
/*----------事前准备---------*/
inline void init()
{
memset(head,-1,sizeof(head));
tot=0;
}
inline void add(int u,int v)
{
e[tot].to=v,e[tot].next=head[u],head[u]=tot++;
}
/*---------树形dp----------*/
inline int dfs(int u,int f,int &ans)
{
fa[u]=f;
int now=0;
for(int i=head[u];i!=-1;i=e[i].next)
{
int to=e[i].to;
if(to==f) continue;
now=max(now,dfs(to,u,ans));
}
if(now<=1)//子树里面的点染不到当前的节点
{
ans++;
return k[u];
}
k[f]=max(k[f],k[u]-1);//如果能够染到的话,就更新父亲节点的染色值,取最大值
return now-1;
}
int main()
{
int x;
while(~scanf("%d",&n))
{
init();
for(int i=2;i<=n;i++)
{
scanf("%d",&x);
add(i,x);add(x,i);
}
for(int i=1;i<=n;i++) scanf("%d",&k[i]);
int ans=0;
dfs(1,0,ans);
cout<<ans<<endl;
}
return 0;
}
努力加油a啊,(o)/~