CF 1436D Bandit in a City 树上贪心

原题链接

题意

给定一棵树形结构，每个点上都有一定的人数，人只能往叶子节点跑，根节点有一个土匪，问土匪最少能抓到多少人。

分析

我们只需要让所有的人均匀地跑到叶子节点就可以了，因此对于每个节点都有一个最优方案，也就是当前子树人数/当前子树叶子节点数+(人数%叶子节点 ? 1 : 0)，最后取一个max就可以了，比较水的一道题。

AC Code

#include <bits/stdc++.h>

#define ACM_LOCAL
#define fi first
#define se second
#define pb push_back
using namespace std;
typedef long long ll;
typedef pair<int, int> PII;
const int N = 5e5 + 10, M = 5e5 + 10, INF = 0x3f3f3f3f;
const int MOD = 1e9 + 7;
int n, m, k, cnt;
ll sz[N], ans[N], a[N], res;
vector<int> g[N];

void dfs(int u) {
    ans[u] = a[u];
    if (g[u].empty()) sz[u] = 1;
    for (auto v : g[u]) {
        dfs(v);
        sz[u] += sz[v];
        ans[u] += ans[v];
    }
}

void count(int u) {
    res = max(res, ans[u] / sz[u] + (ans[u] % sz[u] ? 1 : 0));
    for (auto v : g[u]) {
        count(v);
    }
}

void solve() {
    int n; cin >> n;
    for (int i = 2; i <= n; i++) {
        int x; cin >> x;
        g[x].push_back(i);
    }
    for (int i = 1; i <= n; i++) cin >> a[i];
    dfs(1); count(1);
    cout << res << endl;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
#ifdef ACM_LOCAL
    freopen("input", "r", stdin);
    freopen("output", "w", stdout);
#endif
    solve();
    return 0;
}