用C语言写二叉树的WPL计算
时间: 2023-09-30 13:04:19 浏览: 224
好的,我来回答你的问题。
WPL(Weighted Path Length)是一种度量二叉树的方法,它是每个叶子节点的深度乘以该叶子节点的权值之和。WPL 越小,说明该二叉树越紧凑。
下面是用 C 语言计算二叉树 WPL 的代码:
```c
#include <stdio.h>
#include <stdlib.h>
// 二叉树结点结构体
typedef struct TreeNode {
int val; // 结点权值
struct TreeNode* left; // 左子结点指针
struct TreeNode* right; // 右子结点指针
} TreeNode;
// 创建二叉树
TreeNode* createTree() {
int val;
scanf("%d", &val);
if (val == 0) {
return NULL;
}
TreeNode* root = (TreeNode*)malloc(sizeof(TreeNode));
root->val = val;
root->left = createTree();
root->right = createTree();
return root;
}
// 计算叶子结点的 WPL
int dfs(TreeNode* root, int depth) {
if (root == NULL) {
return 0;
}
if (root->left == NULL && root->right == NULL) { // 叶子结点
return depth * root->val;
}
return dfs(root->left, depth + 1) + dfs(root->right, depth + 1);
}
// 计算二叉树的 WPL
int getWPL(TreeNode* root) {
return dfs(root, 0);
}
int main() {
printf("请输入二叉树的先序遍历序列,其中 0 表示空结点:\n");
TreeNode* root = createTree();
printf("二叉树的 WPL 为:%d\n", getWPL(root));
return 0;
}
```
这段代码中,我们首先定义了一个结构体 `TreeNode`,表示二叉树结点。然后,我们通过 `createTree` 函数创建了一棵二叉树。在 `dfs` 函数中,我们使用深度优先搜索的方法计算每个叶子结点的 WPL。最后,我们通过 `getWPL` 函数计算整个二叉树的 WPL。
希望可以帮助到你!
阅读全文