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

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

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

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

C++算法入门练习——树的带权路径长度_第1张图片

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

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

完整代码如下:

#include 
#include 
using namespace std;

struct node{
	int height = 0;
	int weight;
	vector 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>n;
	for(int i=0;i>nodes[i].weight;
	}
	for(int i=0;i>k;
		int temp;
		for(int j=0;j>temp;
			nodes[i].child.push_back(temp);
		}
	}
	PreOrderTraverse(0);
	cout<

 

你可能感兴趣的:(算法,c++,数据结构,深度优先)