C++算法入门练习——树的带权路径长度

现有一棵n个结点的树（结点编号为从0到n-1，根结点为0号结点），每个结点有各自的权值w。

1. 结点的路径长度是指，从根结点到该结点的边数；
2. 结点的带权路径长度是指，结点权值乘以结点的路径长度；
3. 树的带权路径长度是指，树的所有叶结点的带权路径长度之和。

求这棵树的带权路径长度。

解题思路：每个节点存储weight和height值。

然后利用遍历，当遍历到叶子节点时候，sumweight加上当前节点的带权路径值。

完整代码如下：

#include <iostream>
#include <vector>
using namespace std;

struct node{
	int height = 0;
	int weight;
	vector<int> child;
} nodes[51];
bool flag = false;
int sumweight = 0;

void PreOrderTraverse(int root){
	if(root == -1){
		return ;
	}
	if(nodes[root].child.size()==0){
		sumweight += nodes[root].weight*nodes[root].height;
	} 
	for(int i=0;i<nodes[root].child.size();i++){
		nodes[nodes[root].child[i]].height = nodes[root].height+1;
		PreOrderTraverse(nodes[root].child[i]);
	}
}

int main(){
	int n;
	cin>>n;
	for(int i=0;i<n;i++){
		cin>>nodes[i].weight;
	}
	for(int i=0;i<n;i++){
		int k;
		cin>>k;
		int temp;
		for(int j=0;j<k;j++){
			cin>>temp;
			nodes[i].child.push_back(temp);
		}
	}
	PreOrderTraverse(0);
	cout<<sumweight<<endl;
	return 0;
}