[BZOJ1131][POI2008]Sta（树形dp+维护）

题目描述

传送门

题解

先预处理出每个子树的大小以及以1为根时深度和。
每次暴力转移O（1）

代码

#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define LL long long
const int max_n=1e6+5;
const int max_e=max_n*2;
int n,x,y;
int tot,next[max_e],point[max_n],v[max_e];
LL h[max_n],size[max_n],sum,ans;
int ansp;
inline void add(int x,int y){++tot;next[tot]=point[x];point[x]=tot;v[tot]=y;}
inline void dfs(int x,int fa,int dep){
    h[x]=dep; sum+=h[x]; size[x]=1;
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa){
        dfs(v[i],x,dep+1);
        size[x]+=size[v[i]];
      }
}
inline void calc_ans(int x,int fa,LL sum){
    for (int i=point[x];i;i=next[i])
      if (v[i]!=fa){
        LL t=sum-size[v[i]]+(size[1]-size[v[i]]);
        if (t>ans||t==ans&&v[i]<ansp) ans=t,ansp=v[i];
        calc_ans(v[i],x,t);
      }
}
int main(){
    scanf("%d",&n);
    for (int i=1;i<n;++i) scanf("%d%d",&x,&y),add(x,y),add(y,x);
    dfs(1,0,1); ans=sum; ansp=1;
    calc_ans(1,0,sum);
    printf("%d\n",ansp);
}

总结

不要手残！想好了是父亲还是儿子！！！