CF1187E Tree Painting

思路：

树形dp，首先使用dp计算以1为根的时候的最大分数，同时得到各个子树i的最大分数dp[i]。然后利用前面得到的dp数组分别计算以其他每个点作为根的时候的最大分数。

实现：

 1 #include <bits/stdc++.h>
 2 using namespace std;
 3 typedef long long ll;
 4 const int N = 200005;
 5 vector<int> G[N];
 6 ll cnt[N], dp[N];
 7 int n;
 8 void dfs(int u, int p)
 9 {
10     cnt[u] = 1;
11     for (int i = 0; i < G[u].size(); i++)
12     {
13         int t = G[u][i];
14         if (t == p) continue;
15         dfs(t, u);
16         cnt[u] += cnt[t];
17         dp[u] += dp[t];
18     }
19     dp[u] += cnt[u];
20 }
21 void dfs2(int u, int p, ll x, ll& ans)
22 {
23     ans = max(ans, dp[u] - cnt[u] + x + n);
24     for (int i = 0; i < G[u].size(); i++)
25     {
26         int t = G[u][i];
27         if (t == p) continue;
28         dfs2(t, u, dp[u] - cnt[u] - dp[t] + x + n - cnt[t], ans);
29     }
30 }
31 int main()
32 {
33     int a, b;
34     while (cin >> n)
35     {
36         for (int i = 1; i <= n; i++) G[i].clear();
37         memset(cnt, 0, sizeof cnt); memset(dp, 0, sizeof dp);
38         for (int i = 1; i <= n - 1; i++)
39         {
40             cin >> a >> b;
41             G[a].push_back(b); G[b].push_back(a);
42         }
43         dfs(1, 0);
44         ll maxn = dp[1];
45         for (int i = 0; i < G[1].size(); i++)
46         {
47             int t = G[1][i];
48             dfs2(t, 1, dp[1] - dp[t] - cnt[t], maxn);
49         }
50         cout << maxn << endl;
51     }
52     return 0;
53 }