BZOJ 1131: [POI2008]Sta 树形DP

Description

给出一个N个点的树,找出一个点来,以这个点为根的树时,所有点的深度之和最大

Input

给出一个数字N,代表有N个点.N<=1000000 下面N-1条边.

Output

输出你所找到的点,如果具有多个解,请输出编号最小的那个.

Sample Input

8

1 4

5 6

4 5

6 7

6 8

2 4

3 4

Sample Output

7

解题方法： 裸树DP，发现把根从一个点的父亲移到这个节点DP值的变化差值是delta=(n−size[x])−size[x]。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
const int maxn = 1000005;
LL dp[1000005];
int n, ans, edgecnt, head[maxn], dep[maxn], siz[maxn];
void init(){
memset(head, -1, sizeof(head)); edgecnt = 0;
}
struct edge{int v, nxt; } E[2000005];
void addedge(int u, int v){
E[edgecnt].v = v, E[edgecnt].nxt = head[u], head[u] = edgecnt++;
}
void dfs1(int x, int fa){
siz[x] = 1;
dp[x] = dep[x];
for(int i = head[x]; ~i; i = E[i].nxt){
int v = E[i].v;
if(v == fa) continue;
dep[v] = dep[x] + 1;
dfs1(v, x);
dp[x] += dp[v];
siz[x] += siz[v];
}
}
void dfs2(int x, int fa)
{
for(int i = head[x]; ~i; i = E[i].nxt){
int v = E[i].v;
if(v == fa) continue;
dp[v] = dp[x] + n - 2 * siz[v];
dfs2(v, x);
}
}
int main(){
init();
scanf("%d", &n);
for(int i = 1; i < n; i++){
int u, v;
scanf("%d%d", &u, &v);
addedge(u, v);
addedge(v, u);
}
dfs1(1, -1);
dfs2(1, -1);
for(int i = 1; i <= n; i++){
if(dp[i] > dp[ans]) ans = i;
}
return printf("%d\n", ans);
}