黑白树

2019-04-02  本文已影响0人  MHYo_oYHM

链接:https://ac.nowcoder.com/acm/problem/13249
来源:牛客网

时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld

题目描述

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

解决思路

首先,我们将问题简化为一条链(即退化的树),贪心的想法是从叶子节点开始向根部遍历点,一次能遍历的越多越好,但是这样会出现错误,比如:


2019-04-02_011321.png

如果贪心,我们会从5直接涂黑到3,这样的结果是3;然而,最优解为先从4涂黑到1,然后涂黑5,结果为2。
贪心思想的问题在于:一时的贪心可能会错过涂黑路上k值很大的点。

那么,只要把k值向根部转移不就好了吗

首先明确一点,贪心的做法中,如果涂黑的途中遇到更优质的k值,那么可以将k值转移到涂黑结束的点上。

#include <bits/stdc++.h>
using namespace std;
const int N  = 1e5+100;
int n;
vector<int> v[N];
int k[N],fa[N];
int ans=0;
int dfs(int x){
    int now = 0, siz = v[x].size();
    for(int i=0;i<siz;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;
    scanf("%d", &n);
    for(int i=2;i<=n;i++){
        int d;
        scanf("%d", &d);
        v[d].push_back(i);
        fa[i]=d;
    }
    for(int i=1;i<=n;i++)
        scanf("%d", &k[i]);
    dfs(1);
    printf("%d\n",ans);
}


上一篇下一篇

猜你喜欢

热点阅读