Codeforces Round #398 (Div. 2) C. Garland DFS

C. Garland

链接：

http://codeforces.com/contest/767/problem/C

题意：

给你一颗树，其中包含N个点，让你将这颗树剪去两条边，使得变成三部分，要求三部分的权值和相等。

题解：

1、我们设定sum【i】表示以i作为根的子树权值和。

那么过程维护，sum【i】+=sum【v】；

2、如果全部点的权值和不是3的倍数，那么显然无解，否则考虑：

①遇到了一个点u，此时sum【u】=tot/3；那么这个点就可以将其拆除出来。

②将一个点拆出来的同时，设定sum【u】=0；回溯，继续找下一个tot/3

③如果过程中遇到了两个sum【u】=tot；此时存在解。

代码：

#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <functional>
using namespace std;
#define rep(i,a,n) for (int i=a;i<=n;i++)
#define per(i,a,n) for (int i=n;i>=a;i--)
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
typedef vector<int> VI;
typedef long long ll;
typedef pair<int, int> PII;
const ll mod = 1000000007;
// head

const int maxn = 1e6 + 7;
VI tree[maxn];
int val[maxn];
int sum[maxn];
int n, tot, ans1, ans2, root;

void dfs(int u)
{
    sum[u] = val[u];
    for (int i = 0; i<tree[u].size(); i++){
        int v = tree[u][i];
        dfs(v);
        sum[u] += sum[v];
    }
    if (sum[u] == tot){
        if (u == root) return;
        if (ans1 == -1){
            ans1 = u;
            sum[u] = 0;
        }
        else if (ans2 == -1){
            ans2 = u;
            sum[u] = 0;
        }
    }
}

int main()
{
    cin >> n;
    rep(i, 1, n) {
        int a;
        scanf("%d%d", &a, val + i);
        tot += val[i];
        if (a == 0) {
            root = i; 
            continue;
        }
        tree[a].pb(i);
    }
    if (tot % 3) return 0 * printf("-1
");
    tot /= 3;
    ans1 = ans2 = -1;
    dfs(root);
    if (ans1 == -1 || ans2 == -1) cout << -1 << endl;
    else cout << ans1 << " " << ans2 << endl;
    return 0;
}