黑白树

https://ac.nowcoder.com/acm/problem/13249

题意:把整棵树涂黑,每个点都可以往根上涂k[i]个点,选最少的点。
思路:
可以想到叶子节点必须要选,然后一开始觉得没被选的点一定要被选其实是错的,例如该图
图片说明

如果有部分点k很大不选就亏了,这样我们可以通过k[u]=max(k[u],k[son]-1)来更新最优的k,如何判断某点是否被覆盖?用个dp数组就行了,如果!dp[u]那么就选择该点,同时维护个数量就可以啦。
#include<bits/stdc++.h>
using namespace std;
const int N=100010;
int n,k[N];
vector<int>v[N];
int ans;
int dp[N];
void dfs(int u,int fa)
{
    for(auto j:v[u])
    {
        dfs(j,u);
        dp[u]=max(dp[u],dp[j]-1);
        k[u]=max(k[j]-1,k[u]);
    }
    if(!dp[u])
    {
        ans++;
        dp[u]=k[u];
    }

}
int main()
{
    cin>>n;
    for(int i=2;i<=n;i++)
    {
        int x;
        cin>>x;
        v[x].push_back(i);
    }
    for(int i=1;i<=n;i++)
        cin>>k[i];
    dfs(1,0);
    cout<<ans<<endl;
    return 0;
}
全部评论

相关推荐

12-27 22:14
门头沟学院 Java
点赞 评论 收藏
分享
12-23 18:51
中南大学 Java
唉又萌混过关:是不是那种收钱盖实习章的机构?
点赞 评论 收藏
分享
牛马人的牛马人生:500一天吗?香麻了
投递字节跳动等公司6个岗位
点赞 评论 收藏
分享
12-27 22:35
门头沟学院 Java
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务