E - Complicated Computations CodeForces - 1436E

题目链接：https://codeforces.com/problemset/problem/1436/E

 

通读题目，可以用图或者用树对题目进行建模，每个广场是一个节点，人数就是权值，权值是以它为子树的权值和。

观察第一个测试样例，发现是一种刚好均分的场景，即1号广场的人可以均匀地跑到其余广场。

观察第二个测试样例，发现是一种无法均分的场景，总有至少一个广场比其余人数较少的广场多1个人，那么这个广场的人数就是答案了。

则可得到 ans=max(ans, a[i]/root[i]+(a[i]%root[i]? 1:0));

具体见代码注释：

#include <bits/stdc++.h>
#include <iostream>
using namespace std;

long long n, p[200010], root[200010], a[200010];

int main(){

    cin>>n;
    for(int i=1; i<=n; i++)
        root[i]=1; //初始化所有节点，1代表叶节点
    for(int i=2; i<=n; i++){
        cin>>p[i];
        root[p[i]]=0; //更新节点p[i]为非叶节点，0代表非叶节点
    }
    for(int i=1; i<=n; i++){
        cin>>a[i];
    }
    long long ans=-1;
    for(int i=n; i>=1; i--){
        ans=max(ans, a[i]/root[i]+(a[i]%root[i]? 1:0)); //如果可以均分，就是均分的结果，否则加1。加1的原因是因为要尽可能地优化市民的生存，即每个城市只能多分一个，分完为止，则这些城市的人数就是答案。
        root[p[i]]+=root[i]; //更新p[i]有多少个子节点
        a[p[i]]+=a[i]; //更新p[i]的权值和
    }
    cout<<ans<<endl;
    return 0;
}