51nod-1405 树的距离之和

原题链接

1405 树的距离之和
基准时间限制:1 秒 空间限制:131072 KB 分值: 40  难度:4级算法题
 收藏
 关注
给定一棵无根树,假设它有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

以1为根遍历整棵树,cnt[i]表示以i为根的子树节点数目,dis[i]表示以i为根的子树上的节点到i的距离之和

dp[1] = dis[1], 假如1是2的父节点 dp[2] = dp[1] - cnt[2] - dis[2](除了以i为根节点的子树上的节点,其他节点到1的距离之和) + cnt[1] - cnt[2](除了以i为根节点,剩余的节点数) + dis[2]

以此类推

#include 
#include 
#include 
#include 
#include 
#include 
#define maxn 100005
#define MOD 1000000007
using namespace std;
typedef long long ll;

vector v[maxn];
ll cnt[maxn], dp[maxn], dis[maxn];
void dfs1(int j, int f){
	
	cnt[j] = 1;
	for(int i = 0; i < v[j].size(); i++){
		int d = v[j][i];
		if(d != f){
			dfs1(d, j);
			cnt[j] += cnt[d];
			dis[j] += dis[d] + cnt[d];
		}
	}
}
void dfs2(int j, int f){
	
	for(int i = 0; i < v[j].size(); i++){
		int d = v[j][i];
		if(d != f){
			dp[d] = dis[d] + (dp[j] - dis[d] - cnt[d]) + (cnt[1] - cnt[d]);
			dfs2(d, j);
		}
	}
}
int get_Int(){
	int m = 0;
	char ch = getchar();
	while(ch >= '0' && ch <= '9'){
		m = m * 10 + ch - '0';
		ch = getchar();
	}
	return m;
}
int main(){
	
//	freopen("in.txt", "r", stdin);
	int n, a, b;
	
	scanf("%d", &n);
	getchar();
	for(int i = 0; i < n-1; i++){
		a = get_Int();
		b = get_Int();
		v[a].push_back(b);
		v[b].push_back(a);
	}
	dfs1(1, -1);
	dp[1] = dis[1];
	dfs2(1, -1);
	for(int i = 1; i <= n; i++)
	 printf("%I64d\n", dp[i]);
	return 0;
}


你可能感兴趣的:(动态规划)