P3478 [POI2008]STA-Station（sta树 树形dp）

典型的sta树

得出状态方程dp[u]为将u设置为根节点的全部度数和相加，方程dp[v]=dp[u]+n-2*num[v]（num[v]指的是以v为根节点的子树的节点数）

代码：

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <vector>
using namespace std;
const int maxn=2000100;
typedef long long int LL;
int head[maxn]={},next[maxn]={},to[maxn]={},d[maxn]={},ans2;
LL num[maxn]={},dp[maxn]={};
LL n,cnt,ans;
inline void addedge(int u,int v){
    to[++cnt]=v;
    next[cnt]=head[u];
    head[u]=cnt;
}
int dfs(int u,int fa,int dd){
    num[u]=1;dp[1]+=dd;
    for(int i=head[u];i!=-1;i=next[i])if(to[i]!=fa){
        int v=to[i];
        num[u]+=dfs(v,u,dd+1);
    }
    d[u]=dd;
    return num[u];
}
void dfs2(int u,int fa){
    for(int i=head[u];i!=-1;i=next[i])if(to[i]!=fa){
        int v=to[i];
        dp[v]=dp[u]+n-2*num[v];
        dfs2(v,u);
        if(dp[v]>ans){
            ans=dp[v];
            ans2=v;
        }
        else if(dp[v]==ans) ans2=min(ans2,v);
    }
}
int main()
{
    scanf("%d",&n);cnt=0;ans=0;
    memset(head,-1,sizeof(head));
    memset(d,0,sizeof(d));
    memset(num,0,sizeof(num));
    for(int i=1;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        addedge(u,v);addedge(v,u);
    }
    dfs(1,-1,1);
    ans=dp[1];ans2=1;
    dfs2(1,-1);
    printf("%d\n",ans2);
    return 0;
}