DS二叉树--赫夫曼树解码

DS二叉树–赫夫曼树解码

题目描述

已知赫夫曼编码算法和程序，在此基础上进行赫夫曼解码

可以增加一个函数：int Decode(const string codestr, char txtstr[]);//输入编码串codestr，输出解码串txtstr

该方法如果解码成功则返回1，解码失败则返回-1，本程序增加宏定义ok表示1，error表示-1

赫夫曼解码算法如下：

定义指针p指向赫夫曼树结点，指针i指向编码串，定义ch逐个读取编码串的字符

初始操作包括读入编码串str，设置p指向根结点，i为0表示指向串首，执行以下循环：

1）取编码串的第i个字符放入ch

2）如果ch是字符0，则p跳转到左孩子；如果ch是字符1，则p跳转到右孩子；如果ch非0非1，表示编码串有错误，报错退出

3）如果p指的结点是叶子，输出解码字符，p跳回根结点，i++，跳步骤1

4）如果p指的结点不是叶子且i未到编码串末尾，i++，跳步骤1

5）如果p指的结点不是叶子且i到达编码串末尾，报错退出

当i到达编码串末尾，解码结束。

输入

第一行先输入n，表示有n个叶子

第二行输入n个权值，权值全是小于1万的正整数

第三行输入n个字母，表示与权值对应的字符

第四行输入k，表示要输入k个编码串

第五行起输入k个编码串

输出

每行输出解码后的字符串，如果解码失败直接输出字符串“error”，不要输出部分解码结果

输入样例：

5
15 4 4 3 2
A B C D E
3
11111
10100001001
00000101100

输出样例：

AAAAA
ABEAD
error

参考代码：

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>

using namespace std;

struct Node {
    int weight;
    char data;
    Node *left;
    Node *right;

    Node(int w, char d) : weight(w), data(d), left(nullptr), right(nullptr) {}
};

struct Compare {
    bool operator()(Node *left, Node *right) {
        return (left->weight == right->weight) ? left->data > right->data : left->weight > right->weight;
    }
};

Node *buildHuffmanTree(const vector<int> &weights, const vector<char> &data) {
    priority_queue<Node *, vector<Node *>, Compare> pq;
    for (size_t i = 0; i < data.size(); ++i) {
        pq.push(new Node(weights[i], data[i]));
    }
    while (pq.size() > 1) {
        Node *left = pq.top(); pq.pop();
        Node *right = pq.top(); pq.pop();
        Node *parent = new Node(left->weight + right->weight, '*');
        parent->left = left;
        parent->right = right;
        pq.push(parent);
    }
    return pq.top();
}

string decodeHuffman(Node *root, const string &codestr) {
    string decodedString;
    Node *current = root;

    for (char c : codestr) {
        if (c == '0') current = current->left;
        else if (c == '1') current = current->right;

        if (!current->left && !current->right) {
            decodedString += current->data;
            current = root;
        }
    }

    if (current != root) decodedString = "error";
    return decodedString;
}

int main() {
    int n;
    cin >> n;

    vector<int> weights(n);
    vector<char> dates(n);

    for (int i = 0; i < n; ++i) cin >> weights[i];
    for (int i = 0; i < n; ++i) cin >> dates[i];

    Node *root = buildHuffmanTree(weights, dates);

    int k;
    cin >> k;
    while (k--) {
        string codestr;
        cin >> codestr;
        cout << decodeHuffman(root, codestr) << endl;
    }
    return 0;
}