[POI2008]STA-Station
阿新 • • 發佈:2019-01-06
很明顯而簡單的樹形$dp$:
設以節點$v$為根時的深度和
我們可以先處理出以$1$為根的深度和$f[1]$,那麼我們怎麼樣才能不$dfs n$次來求出以其他點為根的深度和呢
我們考慮現在節點為$u$,子節點為$v$,那麼當$v$為根的時候,$f[v]=f[u]+n-2*siz[v]$($size[v]$為$v$子樹大小),為什麼呢,畫張圖,我們把$v$當做根,就是在把$u$當做根的基礎上,$v$的子樹深度都減少了$1$,總共減少了$siz[v]$,而其他節點的深度都$+1$,總共加了$n-siz[v]$
程式碼:
#include<iostream> #include<cstdio> #define N 1000007 #define ll long long using namespace std; struct Edge { int to,nxt; }edge[N<<1]; int n,cnt; int head[N],siz[N],dep[N]; ll f[N]; void Add(int u,int v) { edge[++cnt]=(Edge){v,head[u]}; head[u]=cnt; } void Dfs(int u,int fa) { siz[u]=1; dep[u]=dep[fa]+1; f[1]+=dep[u]; for(int i=head[u];i;i=edge[i].nxt) { int v=edge[i].to; if(v==fa) continue; Dfs(v,u); siz[u]+=siz[v]; } } void Dp(int u,int fa) { for(int i=head[u];i;i=edge[i].nxt) { int v=edge[i].to; if(v==fa) continue; f[v]=f[u]+n-2*siz[v]; Dp(v,u); } } int main() { scanf("%d",&n); for(int i=1;i<n;++i) { int u,v; scanf("%d%d",&u,&v); Add(u,v); Add(v,u); } Dfs(1,0); Dp(1,0); ll ans=0,maxn=0; for(int i=1;i<=n;++i) if(maxn<f[i]) ans=i,maxn=f[i]; printf("%lld",ans); return 0; }