Huffman編碼

Huffman編碼是一種壓縮編碼柳爽,我們可以借助Huffman樹來完成這個(gè)任務(wù)渠抹。
算法的內(nèi)容可以參考網(wǎng)上的介紹息罗,這里只是使用C++的方式實(shí)現(xiàn)這個(gè)算法芋齿,并且給出了一個(gè)簡單的例子進(jìn)行測(cè)試啸如。

本文涉及到了STL庫的最新知識(shí)侍匙,比如auto關(guān)鍵字,智能指針等等叮雳,總之想暗,最新的STL11庫讓我更愛C++這么語言(C語言雖然簡單,但是C語言還是停留在最初的設(shè)計(jì)的方法上帘不,并沒有提供其他的更方便的方式说莫,比如相關(guān)的容器,不過還好我們有了C++寞焙,不過C++太大了储狭,不太容易學(xué))

代碼如下:

#include <iostream>
#include <vector>
#include <algorithm>
#include <map>
#include <memory>
using namespace std;

struct s_symbol_info
{
    char ch;
    int  w;

    s_symbol_info(char c, int n) :  ch(c), w(n) {}
};

struct huffman_node
{
    typedef shared_ptr<huffman_node> sptr_huffman_node_in;

    // pos
    sptr_huffman_node_in lchild;
    sptr_huffman_node_in rchild;

    // value
    char ch;

    huffman_node() : ch('\0')    { InitPos(); }
    huffman_node(char c) : ch(c) { InitPos(); }
    ~huffman_node() {}

    void InitPos() {
        lchild.reset();
        rchild.reset();
    }
};
typedef shared_ptr<huffman_node> sptr_huffman_node;
typedef shared_ptr<huffman_node> huffman_tree;

struct heap_node
{
    // weight
    int weight;

    // value
    sptr_huffman_node pNode;

    heap_node(int w, sptr_huffman_node p) : weight(w), pNode(p){}
};
typedef shared_ptr<heap_node> sptr_heap_node;
struct s_comp_sptr_heap_node
{
    bool operator() (sptr_heap_node e1, sptr_heap_node e2) {
        return e1->weight > e2->weight;
    }
}CompSPtrHeapNode;

huffman_tree create_huffman_tree(const vector<s_symbol_info> & vecSymbolInfo)
{
    huffman_tree tree;

    vector<sptr_heap_node> vecHeapNode;
    for (auto itr=vecSymbolInfo.begin(); itr!=vecSymbolInfo.end(); ++itr) {
        vecHeapNode.push_back(make_shared<heap_node>(itr->w, make_shared<huffman_node>(itr->ch)));
    }
    make_heap(vecHeapNode.begin(), vecHeapNode.end(), CompSPtrHeapNode);

    while (vecHeapNode.size() > 1) {
        pop_heap(vecHeapNode.begin(), vecHeapNode.end(), CompSPtrHeapNode);
        auto & eTmp1 = vecHeapNode.back();
        auto e1 = eTmp1;
        vecHeapNode.pop_back();

        pop_heap(vecHeapNode.begin(), vecHeapNode.end(), CompSPtrHeapNode);
        auto & eTmp2 = vecHeapNode.back();
        auto e2 = eTmp2;
        vecHeapNode.pop_back();

        auto e = make_shared<heap_node>(e1->weight+e2->weight, make_shared<huffman_node>());

        vecHeapNode.push_back(e);
        push_heap(vecHeapNode.begin(), vecHeapNode.end(), CompSPtrHeapNode);

        e->pNode->lchild  = e1->pNode;
        e->pNode->rchild  = e2->pNode;
    }

    if (!vecHeapNode.empty()) {
        tree = vecHeapNode[0]->pNode;
    }

    return tree;
}

void build_huffman_code(sptr_huffman_node pNode, string & strHuffCode, map<char, string> & mapHuffCode)
{
    sptr_huffman_node lchild = pNode->lchild;
    sptr_huffman_node rchild = pNode->rchild;

    if (lchild == nullptr && rchild == nullptr) {
        mapHuffCode[pNode->ch] = strHuffCode;
        return ;
    }

    strHuffCode += "0";
    build_huffman_code(lchild, strHuffCode, mapHuffCode);
    strHuffCode.pop_back();

    strHuffCode += "1";
    build_huffman_code(rchild, strHuffCode, mapHuffCode);
    strHuffCode.pop_back();

}

map<char, string> build_huffman_code(huffman_tree root)
{
    map<char, string> mapHuffCode;

    string strHuffCode;

    build_huffman_code(root, strHuffCode, mapHuffCode);

    return mapHuffCode;
}

string huffman_code(string str, const map<char, string> & mapHuffCode)
{
    string rt;

    for(int i=0; i<str.size(); ++i) {
        rt += mapHuffCode.at(str[i]);
    }

    return rt;
}

string huffman_decode(string str, huffman_tree root)
{
    string rt;
    auto pNode = root;
    for (int i=0; i<str.size(); ++i) {
        if (str[i] == '0') {
            pNode = pNode->lchild;
        } else {
            pNode = pNode->rchild;
        }

        if (pNode->ch != '\0') {
            rt.push_back(pNode->ch);
            pNode = root;
        }
    }
    return rt;
}

int main()
{
    vector<s_symbol_info> vecSymbolInfo = {
            s_symbol_info( 'a' , 45 ),
            s_symbol_info( 'b' , 13 ),
            s_symbol_info( 'c' , 12 ),
            s_symbol_info( 'd' , 16 ),
            s_symbol_info( 'e' , 9  ),
            s_symbol_info( 'f' , 5  ),
    };

    cout << "Input:";
    for (auto itr=vecSymbolInfo.begin(); itr!=vecSymbolInfo.end(); ++itr) {
        cout << itr->ch << ":" << itr->w << " ";
    }
    cout << endl;

    auto huffTree = create_huffman_tree(vecSymbolInfo);

    auto mapHuffCode = build_huffman_code(huffTree);

    cout << "Output Huffman Code:" << endl;
    for (auto itr=mapHuffCode.begin(); itr!=mapHuffCode.end(); ++itr) {
        cout << "\t" << itr->first << ":" << itr->second << endl;
    }

    string strTest = "abaadcef";
    cout << "Input Test String:" << strTest << endl;

    string strHuffCode = huffman_code(strTest, mapHuffCode);
    cout << "OutHuffmanCode:";
    cout << strHuffCode << endl;
    cout << "OutHuffmanDecode:" << huffman_decode(strHuffCode, huffTree) << endl;
    return 0;
}

輸出如下:

Input:a:45 b:13 c:12 d:16 e:9 f:5 
Output Huffman Code:
    a:0
    b:101
    c:100
    d:111
    e:1101
    f:1100
Input Test String:abaadcef
OutHuffmanCode:01010011110011011100
OutHuffmanDecode:abaadcef

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末互婿,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子辽狈,更是在濱河造成了極大的恐慌慈参,老刑警劉巖,帶你破解...
    沈念sama閱讀 219,110評(píng)論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件刮萌,死亡現(xiàn)場(chǎng)離奇詭異驮配,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)着茸,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,443評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門壮锻,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人元扔,你說我怎么就攤上這事躯保。” “怎么了澎语?”我有些...
    開封第一講書人閱讀 165,474評(píng)論 0 356
  • 文/不壞的土叔 我叫張陵途事,是天一觀的道長。 經(jīng)常有香客問我擅羞,道長尸变,這世上最難降的妖魔是什么? 我笑而不...
    開封第一講書人閱讀 58,881評(píng)論 1 295
  • 正文 為了忘掉前任减俏,我火速辦了婚禮召烂,結(jié)果婚禮上,老公的妹妹穿的比我還像新娘娃承。我一直安慰自己奏夫,他們只是感情好,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,902評(píng)論 6 392
  • 文/花漫 我一把揭開白布历筝。 她就那樣靜靜地躺著酗昼,像睡著了一般。 火紅的嫁衣襯著肌膚如雪梳猪。 梳的紋絲不亂的頭發(fā)上麻削,一...
    開封第一講書人閱讀 51,698評(píng)論 1 305
  • 那天,我揣著相機(jī)與錄音春弥,去河邊找鬼呛哟。 笑死,一個(gè)胖子當(dāng)著我的面吹牛匿沛,可吹牛的內(nèi)容都是我干的扫责。 我是一名探鬼主播,決...
    沈念sama閱讀 40,418評(píng)論 3 419
  • 文/蒼蘭香墨 我猛地睜開眼逃呼,長吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼公给!你這毒婦竟也來了借帘?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,332評(píng)論 0 276
  • 序言:老撾萬榮一對(duì)情侶失蹤淌铐,失蹤者是張志新(化名)和其女友劉穎,沒想到半個(gè)月后蔫缸,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體腿准,經(jīng)...
    沈念sama閱讀 45,796評(píng)論 1 316
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,968評(píng)論 3 337
  • 正文 我和宋清朗相戀三年拾碌,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了吐葱。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片。...
    茶點(diǎn)故事閱讀 40,110評(píng)論 1 351
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡校翔,死狀恐怖弟跑,靈堂內(nèi)的尸體忽然破棺而出,到底是詐尸還是另有隱情防症,我是刑警寧澤孟辑,帶...
    沈念sama閱讀 35,792評(píng)論 5 346
  • 正文 年R本政府宣布,位于F島的核電站蔫敲,受9級(jí)特大地震影響饲嗽,放射性物質(zhì)發(fā)生泄漏。R本人自食惡果不足惜奈嘿,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,455評(píng)論 3 331
  • 文/蒙蒙 一貌虾、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧裙犹,春花似錦尽狠、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 32,003評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至盗似,卻和暖如春哩陕,著一層夾襖步出監(jiān)牢的瞬間,已是汗流浹背赫舒。 一陣腳步聲響...
    開封第一講書人閱讀 33,130評(píng)論 1 272
  • 我被黑心中介騙來泰國打工悍及, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留,地道東北人接癌。 一個(gè)月前我還...
    沈念sama閱讀 48,348評(píng)論 3 373
  • 正文 我出身青樓心赶,卻偏偏與公主長得像,于是被迫代替她去往敵國和親缺猛。 傳聞我的和親對(duì)象是個(gè)殘疾皇子缨叫,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,047評(píng)論 2 355

推薦閱讀更多精彩內(nèi)容