codeM 美团 b轮 A题 黑白树

一棵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变黑

输出描述:
一个数表示最少操作次数

input
5
1
2
3
2
1 2 3 2 1
output
3

最大的错误就是树形dp一直觉得 dp是代表最小操作数 其实dp往控制范围想比较好,这个能得到直接的父子结点的递推关系 然后 就是不断的更新父结点,贪心的决定是否要选,而且每次从下往上更新直接父结点的最优解。一旦now不足,要更换now的时候,就用最优解来更换。这样能保证所有的非叶子结点都是最优解。这样的now也会一直是最优。

#include <bits/stdc++.h>
using namespace std;
int n;
const int N  = 1e5+100;
vector<int> v[N];
int k[N],fa[N];
int ans=0;
int dfs(int x){
    int now=0;
    for(int i=0;i<v[x].size();i++)
    {
        now=max(now,dfs(v[x][i]));
    }
    if(now<=0)
    {
        ans++;
        return k[x]-1;
    }
    k[fa[x]]=max(k[fa[x]],k[x]-1);
    return now-1;
}

int main()
{
    int n;
    cin>>n;
    for(int i=2;i<=n;i++)
    {
        int d;
        cin>>d;
        v[d].push_back(i);
        fa[i]=d;
    }
    for(int i=1;i<=n;i++)
        cin>>k[i];
    dfs(1);
    printf("%d\n",ans );
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值