傳送門
一開始把題意讀錯了。
我們dfs時對于邊$(p,fa)$，計算出以$p$為根的子樹對子樹外連通塊的貢獻，然后加上漏加的貢獻。
這樣算出來是答案的兩倍。
因為相當于$dis(u,v)$和$dis(v,u)$的貢獻都被計入了答案。
代碼：

#include<bits/stdc++.h> #define N 200005 #define ll long long using namespace std; inline int read(){int ans=0;char ch=getchar();while(!isdigit(ch))ch=getchar();while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();return ans; } int n,cnt,first[N],tot[2],siz[N]; ll ans=0; struct edge{int v,next;}e[N<<1]; inline void add(int u,int v){e[++cnt].v=v,e[cnt].next=first[u],first[u]=cnt;} inline void dfs(int p,int fa,int op){siz[p]=1,++tot[op];for(int i=first[p];i;i=e[i].next){int v=e[i].v;if(v==fa)continue;dfs(v,p,op^1);siz[p]+=siz[v];}ans+=1ll*siz[p]*(1ll*(n-siz[p])); } int main(){n=read();for(int i=1;i<n;++i){int u=read(),v=read();add(u,v),add(v,u);}dfs(1,0,0);cout<<(ans+1ll*tot[0]*(1ll*tot[1])>>1); return 0; }

總結

以上是生活随笔為你收集整理的2018.10.04 codeforces1060E. Sergey and Subway（树形dp）的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。