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);}