CodeM美团B轮A题 黑白树 【贪心】

题目地址

思路 : 题意应该好懂. 思路就是贪心, 每次都去选择叶结点去更新, 因为叶结点始终都会去更新的, 所以我们就贪心的取叶结点去染色， 然后把叶子和它上面k-1个给染了, 继续遍历那k-1个, 如果没染染上就好了, 如果染上了，它就把它的k值减一赋给他的父亲 ，他的父亲就在他本身的k和他儿子赋给他的取个max, 因为有可能染他儿子更优, 所以对它上面那个点来说就是k-1, 我们只是去贪心的选择, 而实际上我们是可以尽量选择最优的来做.

AC Code (邻接表比vector更快, 防止卡常, 还是用邻接表好一点吧)

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
int fa[maxn],k[maxn];
int vis[maxn];
int res=0;
struct node
{
    int next;
    int u;
}s[maxn];
int head[maxn];
int cnt=0;
void add(int u,int v)
{
    s[cnt].next = v;
    s[cnt].u= head[u];
    head[u] = cnt++;
}
void dfs(int x)
{
    for(int i=head[x]; ~i ;i = s[i].u)
        dfs(s[i].next);
    if(vis[x] == 1){
        if(x != 1){
            k[fa[x]] = max(k[x]-1,k[fa[x]]);
        }
    }
    else{
        int h = k[x];
        int cur = x;
        res++;
        while(h){
            h--;
            vis[cur] = 1;
            if(cur == 1) break;
            cur = fa[cur];
        }
    }
}

int main()
{
    int n;
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    for(int i=2;i<=n;i++){
        int u;scanf("%d",&u);
        fa[i] = u;
        add(u,i);
    }
    for(int i=1;i<=n;i++)
        scanf("%d",&k[i]);
    dfs(1);
    printf("%d",res);
}