c语言编写二叉树的WPL计算
时间: 2023-10-25 08:24:48 浏览: 251
二叉树构造与遍历.docx
好的,您可以通过以下的C语言代码来计算二叉树的WPL(带权路径长度):
```c
#include <stdio.h>
#include <stdlib.h>
struct node {
int data;
int weight;
struct node *left;
struct node *right;
};
// 创建一个新的二叉树节点
struct node *new_node(int data, int weight) {
struct node *new = (struct node *)malloc(sizeof(struct node));
new->data = data;
new->weight = weight;
new->left = NULL;
new->right = NULL;
return new;
}
// 计算二叉树的WPL
int calc_wpl(struct node *root, int depth) {
// 如果二叉树为空,则返回0
if (root == NULL) {
return 0;
}
// 如果是叶子节点,则返回该节点的权重乘以深度
if (root->left == NULL && root->right == NULL) {
return root->weight * depth;
}
// 否则,递归计算左子树和右子树的WPL之和
return calc_wpl(root->left, depth + 1) + calc_wpl(root->right, depth + 1);
}
int main() {
// 创建一个二叉树
struct node *root = new_node('+', 0);
root->left = new_node('*', 2);
root->left->left = new_node('a', 4);
root->left->right = new_node('b', 5);
root->right = new_node('*', 3);
root->right->left = new_node('c', 2);
root->right->right = new_node('d', 1);
// 计算二叉树的WPL
int wpl = calc_wpl(root, 0);
printf("该二叉树的WPL为:%d\n", wpl);
return 0;
}
```
以上的代码中,我们通过定义一个结构体来表示二叉树的节点,其中包括节点的值和权重以及左右子树的指针。然后,我们通过递归的方式计算二叉树的WPL,具体步骤如下:
1. 如果二叉树为空,则返回0;
2. 如果是一个叶子节点,则返回该节点的权重乘以深度;
3. 否则,递归计算左子树和右子树的WPL之和。
最后,我们通过调用`calc_wpl`函数来计算二叉树的WPL。以上的代码中,我们使用了一个简单的例子来测试计算WPL的代码,其中'+','-','*','/'等符号表示运算符号,而'a','b','c','d'等字母则表示操作数。您可以根据自己的需要修改以上的例子来创建不同的二叉树。
阅读全文