codeforces(D. Bandit in a City)思维

题目链接
给一个有向图,就是一棵树,1为根节点,每个点都有一些权值,将所有点的权值都分配到所有叶子节上,怎样分配使得权值最大的叶子节点的权值最小。

dfs,对每个结点,计算以其为根的叶子节点的个数,若想满足题意,权值肯定是平均分给叶子节最好,但是有可能有一些节点本身权值就很大,如下图,因为4号节点权值过大,3号节点就算将权值全分给8、9、10也抵不上.
在这里插入图片描述

最后计算结果还是按照将以该节点为根的子树的权值和平均分配来算,如果出现上图的情况,在计算到节点4时会更新结果。
代码:

//#pragma GCC optimize("Ofast")
//#pragma GCC target("avx,avx2,fma")
//#pragma GCC optimization ("unroll-loops")
#include <bits/stdc++.h>
#define pb push_back
#define ll long long
#define IOS std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
inline int read() {
    
    
    char c = getchar(); int x = 0, f = 1;
    while (c < '0' || c > '9') {
    
    if (c == '-')f = -1; c = getchar();}
    while (c >= '0' && c <= '9') {
    
    x = x * 10 + c - '0'; c = getchar();}
    return x * f;
}
using namespace std;

ll n;
ll p[200005],a[200005];
ll val[200005],c[200005];
ll max_val=0;
vector<int> edg[200005];
void dfs(ll u)
{
    
    
    bool flag=false;
    for(int i=0;i<edg[u].size();i++)
    {
    
    
        ll v=edg[u][i];
        dfs(v);
        flag=true;
        c[u]+=c[v];
        a[u]+=a[v];
    }
    if(!flag)//叶子节点
    {
    
    
        c[u]=1;
        return;
    }
}
int main()
{
    
    
    scanf("%lld",&n);
    for(int i=2;i<=n;i++)
    {
    
    scanf("%lld",&p[i]);
    edg[p[i]].push_back(i);}
    for(int i=1;i<=n;i++)
    {
    
    scanf("%lld",&a[i]);}
    dfs(1);
    for(int i=1;i<=n;i++)
    {
    
    
        max_val=max(max_val,a[i]/c[i]+(a[i]%c[i]!=0));
    }
    printf("%lld\n",max_val);
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_43781431/article/details/112566301