Codeforces - 1098A - Sum in the tree

地址

https://codeforces.com/contest/1098/problem/A

原文地址

https://www.lucien.ink/archives/400/

题意

给出一棵有点权的树,跟节点深度为 $1$ ,现在你只知道深度为奇数的点到数根的路经权值和,让你给所有点都分配一个非负的权值,满足所给和,且所有点权的和最小。如果不存在输出 $-1$ 。

题解

对于权值和未知的节点,让他们的点权最大即可。

代码

https://pasteme.cn/4440

#include <bits/stdc++.h>
typedef long long ll;
const int maxn = int(1e5) + 7;
std::vector<int> edge[maxn];
int n, dep[maxn];
ll sum[maxn], ans;
void dfs(int u, int pre = 0) {
    dep[u] = dep[pre] + 1;
    if (dep[u] & 1) for (int v : edge[u]) dfs(v, u);
    else {
        if (edge[u].empty()) sum[u] = sum[pre];
        else {
            sum[u] = 0x3f3f3f3f;
            for (int v : edge[u]) {
                dfs(v, u);
                sum[u] = std::min(sum[u], sum[v]);
            }
        }
    }
}
void calc(int u, int pre = 0) {
    if (sum[u] < sum[pre]) {
        puts("-1");
        exit(0);
    }
    ans += sum[u] - sum[pre];
    for (int v : edge[u])
        calc(v, u);
}
int main() {
    scanf("%d", &n);
    for (int i = 2, pre; i <= n; i++) scanf("%d", &pre), edge[pre].push_back(i);
    for (int i = 1; i <= n; i++) scanf("%lld", sum + i);
    dfs(1);
    calc(1);
    printf("%lld\n", ans);
    return 0;
}

发表留言

人生在世,错别字在所难免,无需纠正。