第一行包含一个正整数n (n <= 100000),表示节点个数。 后面(n - 1)行,每行两个整数表示树的边。
每行一个整数,第i(i = 1,2,...n)行表示所有节点到第i个点的距离之和。
4 1 2 3 2 4 2
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; }