1. 程式人生 > >51nod 1405 樹的距離之和 樹形dp

51nod 1405 樹的距離之和 樹形dp

turn amp plus lin src comment code clu 子節點

1405 樹的距離之和 基準時間限制:1 秒 空間限制:131072 KB 技術分享 收藏 技術分享 關註 給定一棵無根樹,假設它有n個節點,節點編號從1到n, 求任意兩點之間的距離(最短路徑)之和。 Input
第一行包含一個正整數n (n <= 100000),表示節點個數。
後面(n - 1)行,每行兩個整數表示樹的邊。
Output
每行一個整數,第i(i = 1,2,...n)行表示所有節點到第i個點的距離之和。
Input示例
4
1 2
3 2
4 2
Output示例
5
3
5
5

思路:dp[i]表示以1為根,以i為子樹的所有子節點到i的最短距離之和;

   dfs遍歷求dp數組,和以i為子樹的節點數和si;

   dfs2求總父親節點來的價值;

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include<iostream>
#include<cstdio>
#include<cmath>
#include<string>
#include<queue>
#include<algorithm>
#include<stack>
#include<cstring>
#include<vector>
#include
<list> #include<bitset> #include<set> #include<map> #include<time.h> using namespace std; #define LL long long #define bug(x) cout<<"bug"<<x<<endl; const int N=1e5+10,M=1e6+10,inf=1e9+10,MOD=1e9+7; const LL INF=1e18+10,mod=1e9+7; const double eps=(1e-8
),pi=(4*atan(1.0)); struct is { int v,nex; }edge[N<<1]; int head[N],edg; void add(int u,int v) { edge[++edg]=(is){v,head[u]}; head[u]=edg; } LL ans[N],a[N]; int si[N]; void dfs(int u,int fa) { si[u]=1; for(int i=head[u];i;i=edge[i].nex) { int v=edge[i].v; if(v==fa)continue; dfs(v,u); si[u]+=si[v]; a[u]+=a[v]+si[v]; } } int n; void dfs2(int u,int fa,LL now) { ans[u]=a[u]+now; for(int i=head[u];i;i=edge[i].nex) { int v=edge[i].v; if(v==fa)continue; dfs2(v,u,now+a[u]-a[v]-si[v]+n-si[v]); } } 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); dfs2(1,0,0); for(int i=1;i<=n;i++) printf("%lld\n",ans[i]); return 0; }

51nod 1405 樹的距離之和 樹形dp