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

来源：互联网发布：ubuntu登陆无限循环编辑：程序博客网时间：2024/05/16 00:43

题目描述

传送门

题解

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

代码

#include<iostream>#include<cstring>#include<cstdio>using namespace std;#define LL long longconst 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);}

总结

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

0 0