题目描述

你有n个节点的一棵树,n的量级是1e6。
后面存在n-1行,第i行代表第i个节点的父节点是谁,并且给出第i个节点的权值,父节点是0的节点就是根。
现在需要你找出一种方案,把两条连接的边切断,把这棵树分成三颗树,并且三棵树的权值之和相等。是否存在这种切法。
如果不存在的话,需要输出-1。

Solution

第一步比较容易的特判就是节点权值之和如果不是3的倍数,那就不存在分法可以分成三等分,直接-1。
第二步的情况,我们把权值之和除以三,遍历这颗树,把子树权值累加,当加到一个节点的权值等于 sum / 3的时候,把这个节点分隔开,也就是把这个节点的权值改成0,不连接在父节点之下了。
最后只要特判是否分出三个点来了即可。

#include <bits/stdc++.h>
using namespace std;
#define js ios::sync_with_stdio(false);cin.tie(0); cout.tie(0)
#define all(__vv__) (__vv__).begin(), (__vv__).end()
#define endl "\n"
#define pai pair<int, int>
#define ms(__x__,__val__) memset(__x__, __val__, sizeof(__x__))
typedef long long ll; typedef unsigned long long ull; typedef long double ld;
inline ll read() { ll s = 0, w = 1; char ch = getchar(); for (; !isdigit(ch); ch = getchar()) if (ch == '-') w = -1; for (; isdigit(ch); ch = getchar())    s = (s << 1) + (s << 3) + (ch ^ 48); return s * w; }
inline void print(ll x, int op = 10) { if (!x) { putchar('0'); if (op)    putchar(op); return; }    char F[40]; ll tmp = x > 0 ? x : -x;    if (x < 0)putchar('-');    int cnt = 0;    while (tmp > 0) { F[cnt++] = tmp % 10 + '0';        tmp /= 10; }    while (cnt > 0)putchar(F[--cnt]);    if (op)    putchar(op); }
inline ll gcd(ll x, ll y) { return y ? gcd(y, x % y) : x; }
ll qpow(ll a, ll b) { ll ans = 1;    while (b) { if (b & 1)    ans *= a;        b >>= 1;        a *= a; }    return ans; }    ll qpow(ll a, ll b, ll mod) { ll ans = 1; while (b) { if (b & 1)(ans *= a) %= mod; b >>= 1; (a *= a) %= mod; }return ans % mod; }
inline int lowbit(int x) { return x & (-x); }
const int dir[][2] = { {0,1},{1,0},{0,-1},{-1,0},{1,1},{1,-1},{-1,1},{-1,-1} };
const int MOD = 1e9 + 7;
const int INF = 0x3f3f3f3f;

const int N = 1e6 + 7;
vector<int> edge[N];
vector<int> ans;
int t[N], val;

void dfs(int u, int fa) {
    for (auto v : edge[u]) {
        dfs(v, u);
        t[u] += t[v];
    }
    if (t[u] == val)
        t[u] = 0, ans.push_back(u);
}

int main() {
    int n = read(), rt = 0, sum = 0;
    for (int i = 1; i <= n; ++i) {
        int x = read();
        t[i] = read();
        sum += t[i];
        if (x)    edge[x].push_back(i);
        else rt = i;
    }
    if (sum % 3)
        return puts("-1"), 0;
    val = sum / 3;
    dfs(rt, 0);
    if (ans.size() < 3)
        puts("-1");
    else
        printf("%d %d\n", ans[0], ans[1]);
    return 0;
}