大致题意:让你求以每个点为中心,所有点到这个中心的距离和。
思路:看了大佬代码,先处理包含每个点的子树的个数,我们当且仅把1当作根结点,然后往下传。然后处理出所有点到1的距离和,
res[u] += num[j]; res[u] += res[j];
我们可以很好的证明出,u以后的子节点到u的距离无非就是多加了一条u->v的·边,所以就是多加了子树的大小。
然后接下来就是求别的点的答案了,可以从图发现,当中心从u变为v的时候 v的子树都不会走u到v这条边了,所以减去子树v的大小,然而上面的节点都会走u到v这条边,答案就加上了(n-v)
#include<iostream>
#include<cstring>
using namespace std;
const int N=100010;
int h[N],e[2*N],idx,ne[N*2];
int n;
typedef long long ll;
int num[N];
ll res[N];
void add(int a,int b)
{
e[idx]=b , ne[idx]=h[a],h[a]=idx++;
}
void dfs(int u,int fa)
{
num[u] = 1;
for(int i=h[u];~i;i=ne[i])
{
int j=e[i];
if(j==fa) continue;
dfs(j,u);
res[u] += num[j];
res[u] += res[j];
num[u]+=num[j];
}
}
void dfs2(int u,int fa)
{
for(int i=h[u];~i;i=ne[i])
{
int j= e[i];
if(j==fa) continue;
res[j]=res[u]-num[j]+n-num[j];
dfs2(j,u);
}
}
int main()
{
memset(h,-1,sizeof h);
cin >> n;
for(int i=1;i<=n-1;i++)
{
int a,b;
cin >> a >> b;
add(a,b);
add(b,a);
}
dfs(1,-1);
dfs2(1,-1);
for(int i=1;i<=n;i++)
cout<<res[i]<<endl;
return 0;
}