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；此时存在解。

代码：

 1 #include <map>
 2 #include <set>
 3 #include <cmath>
 4 #include <queue>
 5 #include <stack>
 6 #include <cstdio>
 7 #include <string>
 8 #include <vector>
 9 #include <cstring>
10 #include <iostream>
11 #include <algorithm>
12 #include <functional>
13 using namespace std;
14 #define rep(i,a,n) for (int i=a;i<=n;i++)
15 #define per(i,a,n) for (int i=n;i>=a;i--)
16 #define pb push_back
17 #define mp make_pair
18 #define all(x) (x).begin(),(x).end()
19 #define fi first
20 #define se second
21 #define SZ(x) ((int)(x).size())
22 typedef vector<int> VI;
23 typedef long long ll;
24 typedef pair<int, int> PII;
25 const ll mod = 1000000007;
26 // head
27 
28 const int maxn = 1e6 + 7;
29 VI tree[maxn];
30 int val[maxn];
31 int sum[maxn];
32 int n, tot, ans1, ans2, root;
33 
34 void dfs(int u)
35 {
36     sum[u] = val[u];
37     for (int i = 0; i<tree[u].size(); i++){
38         int v = tree[u][i];
39         dfs(v);
40         sum[u] += sum[v];
41     }
42     if (sum[u] == tot){
43         if (u == root) return;
44         if (ans1 == -1){
45             ans1 = u;
46             sum[u] = 0;
47         }
48         else if (ans2 == -1){
49             ans2 = u;
50             sum[u] = 0;
51         }
52     }
53 }
54 
55 int main()
56 {
57     cin >> n;
58     rep(i, 1, n) {
59         int a;
60         scanf("%d%d", &a, val + i);
61         tot += val[i];
62         if (a == 0) {
63             root = i; 
64             continue;
65         }
66         tree[a].pb(i);
67     }
68     if (tot % 3) return 0 * printf("-1\n");
69     tot /= 3;
70     ans1 = ans2 = -1;
71     dfs(root);
72     if (ans1 == -1 || ans2 == -1) cout << -1 << endl;
73     else cout << ans1 << " " << ans2 << endl;
74     return 0;
75 }