树的权值深度

时间限制：1秒        内存限制：128M

题目描述

求树的各个节点的权值有一种方法是自上而下统计信息。起初，我们已知根节点s的的权值深度为0。若某个节点x的权值深度为d[x]，x与它的某个子节点y之间的边的权值为z(1<=z<=10000)则它的子节点y的权值深度d[y]=d[x]+z，现在给你一棵树，让你输出每个节点的权值深度。

输入描述

第一行，两个整数n(1<=n<=200000)，s,n表示树的节点的个数，s表示树的根节点的编号。
接下来的n-1行中，每行有三个整数x,y,z，表示x和y有一条边 且权值为z。

输出描述

输出一行，表示每个节点的权值深度，用空格间隔

样例

输入

9 1
1 2 3
1 7 2
1 4 6 
2 8 7
2 5 1212
4 6 10000
4 3 1
3 9 12

输出

0 3 7 6 1215 10006 2 10 19

#include<iostream>
#include<algorithm>
#include<cstring>
#include<string>
#include<cstdio>
using namespace std;
int n,s;
int x,y,z;
const int N=2e5+10;
int head[N],w[2*N],ver[2*N],Next[2*N],vis[N],step[2*N],tot=-1,cnt;
void Add(int x,int y,int z) {
	ver[++tot]=y;
	w[tot]=z;
	Next[tot]=head[x];
	head[x]=tot;
}
void dfs(int x) {
	vis[x]=1;
	for(int i=head[x]; i!=-1; i=Next[i]) {
		int ljd=ver[i];
		if(vis[ljd]==0) {
			step[ljd]=step[x]+w[i];
			dfs(ljd);
		}
	}
}

int main() {
	memset(head,-1,sizeof(head));
	cin>>n>>s;
	for(int i=1; i<=n; i++) {
		cin>>x>>y>>z;
		Add(x,y,z);
		Add(y,x,z);
	}
	step[s]=0;
	dfs(s);
	for(int i=1; i<=n; i++) {
		cout<<step[i]<<" ";
	}
	return 0;
}