用C++代码求哈夫曼树WPL值,要用类的方法
时间: 2024-03-26 12:41:20 浏览: 53
哈夫曼树 的c++代码
4星 · 用户满意度95%
下面是一个使用C++语言,用类的方法实现求哈夫曼树WPL值的示例代码:
```c++
#include <iostream>
#include <queue>
using namespace std;
class TreeNode {
public:
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class HuffmanTree {
private:
priority_queue<TreeNode*, vector<TreeNode*>, cmp> q;
int n;
struct cmp {
bool operator()(TreeNode* a, TreeNode* b) {
return a->val > b->val;
}
};
int getWPL(TreeNode* root, int depth) {
if (root->left == NULL && root->right == NULL) {
return root->val * depth;
}
int wpl = 0;
if (root->left != NULL) {
wpl += getWPL(root->left, depth + 1);
}
if (root->right != NULL) {
wpl += getWPL(root->right, depth + 1);
}
return wpl;
}
public:
HuffmanTree() {
cin >> n;
for (int i = 0; i < n; i++) {
int val;
cin >> val;
TreeNode* node = new TreeNode(val);
q.push(node);
}
while (q.size() > 1) {
TreeNode* node1 = q.top();
q.pop();
TreeNode* node2 = q.top();
q.pop();
TreeNode* newNode = new TreeNode(node1->val + node2->val);
newNode->left = node1;
newNode->right = node2;
q.push(newNode);
}
}
int getWPL() {
TreeNode* root = q.top();
return getWPL(root, 0);
}
};
int main() {
HuffmanTree tree;
int wpl = tree.getWPL();
cout << wpl << endl;
return 0;
}
```
该代码使用了类`HuffmanTree`来实现哈夫曼树的构建和WPL值的计算。在类中,定义了一个私有成员变量`q`,用于存储优先队列;定义了一个私有成员函数`getWPL`,用于递归计算WPL值;定义了一个公有成员函数`getWPL`,用于调用`getWPL`计算WPL值,并返回计算结果。
在`main`函数中,首先创建一个`HuffmanTree`对象,然后调用其`getWPL`函数计算WPL值,并输出结果即可。
阅读全文