Codeforces Round #530 (Div. 2)

最新推荐文章于 2025-12-19 15:31:04 发布

原创最新推荐文章于 2025-12-19 15:31:04 发布 · 89 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法

本文深入探讨了一道编号为D-Suminthetree的算法题，通过将权值分配到树状结构的祖先节点上，采用贪心策略实现最优解。文章详细介绍了算法思路，包括预处理节点权重，计算子节点与父节点间的最小权值，以及处理非法情况的技巧。

RANK ：2252 题数：3

补题：

D - Sum in the tree

思路：贪心把权值放在祖先节点上，预处理每个节点保存他与他儿子中权值最小值即可。

最后会有一些叶子节点依旧为 INF 权值按0算即可，然后其他的权值计算为它 - 它父亲的。

注意判断时候会出现父亲比儿子大的这种非法情况。

#include<bits/stdc++.h>
using namespace std;
#define inf 0x7f7f7f7f
#define ll long long
#define maxn 123456
ll fa[maxn],n,s[maxn],ans;
int main()
{
    scanf("%lld",&n);
    for(int i=2; i<=n; i++)
        scanf("%lld",&fa[i]);
    for(int i=1; i<=n; i++)
    {
        scanf("%lld",&s[i]);
        if(s[i]==-1)s[i]=inf;
    }
    ans=s[1];
    for(int i=2; i<=n; i++)
        s[fa[i]]=min(s[fa[i]],s[i]);
    for(int i=2; i<=n; i++)
    {
        if(s[i]==inf)s[i]=0;
        else
        {
            ans+=s[i]-s[fa[i]];
            if(s[i]-s[fa[i]]<0)
            {
                printf("-1\n");
                return 0;
            }
        }
    }
    printf("%lld\n",ans);
    return 0;
}