Huffman Tree

image.png

image.png

image.png

• 前缀编码
  前缀编码 是指对字符集进行编码时，要求字符集中任一字符的编码都不是其它字符的编码的前缀，例如：设有abcd需要编码表示（其中，a=0、b=10、c=110、d=11,则表示110的前缀可以是c或者da，不唯一）。
• Huffman编码
  用构造哈夫曼树的过程生成的二进制前缀编码。
• Huffman树
  一类带权路径长度最短的树

PTA_7-1修理牧场

image.png

最简单的Huffman树。直接上代码

#include<bits/stdc++.h>
using namespace std;
priority_queue <int, vector<int>, greater<int> > q;
int main(){
    int N, sum = 0, a, b, t;
    cin >> N;
    while(N--){ cin >> t; q.push(t);}
    while(!q.empty()){
        a = q.top(); q.pop();
        if(!q.empty()){
            b = q.top();
            q.pop();
            sum += a + b;
            q.push(a + b);
        }
        else break;
    }
    cout << sum << endl;
    return 0;
}

PTA_7-2哈夫曼编码

 

image.png

输入样例：
7
A 1 B 1 C 1 D 3 E 3 F 6 G 6
4
A 00000
B 00001
C 0001
D 001
E 01
F 10
G 11
A 01010
B 01011
C 0100
D 011
E 10
F 11
G 00
A 000
B 001
C 010
D 011
E 100
F 101
G 110
A 00000
B 00001
C 0001
D 001
E 00
F 10
G 11
输出样例：
Yes
Yes
No
No

 

先上AC代码：

#include<bits/stdc++.h>
#define ll long long
using namespace std;
typedef struct{
    char c, code[70];
    int len;
}D;
D d[70];
int F[300];//记录字符对应频率
map <long long, int> vis[1010]; 
bool cmp2(D x, D y){
    return x.len < y.len;
}
priority_queue <int, vector<int>, greater<int> > q; 
int huffman();
int main()
{
    int N, M, c, f;
    scanf("%d", &N); getchar();
    for(int i = 0; i < N; i++){
        scanf("%c %d", &c, &f); getchar();
        F[c] = f; 
        q.push(f);
    }
    int h = huffman();
    scanf("%d", &M); getchar();
    while(M--){
        int flag = 1;
        for(int i = 0; i < N; i++){
            scanf("%c%s", &d[i].c, d[i].code); getchar();
            d[i].len = strlen(d[i].code);
        }
        //是否前缀编码 
        sort(d, d + N, cmp2);
        for(int i = 0; i < N; i++){
            ll sum = 0;
            for(int j = 0; j < d[i].len; j++){
                sum = sum * 3 + (d[i].code[j] - '0' + 1);//注意前导零 
                if(vis[M][sum]){ flag = 0; break;}
            }
            if(!flag) break;
            vis[M][sum] = 1;
        } 
        if(!flag){
            printf("No\n"); continue;
        }
        //是否最优长度 
        int hh = 0;
        for(int i = 0; i < N; i++) hh += d[i].len * F[d[i].c];
        if(hh == h) printf("Yes\n");
        else printf("No\n");
    }
    return 0;
}
int huffman(){
    int a, b, sum = 0;
    while(!q.empty()){
        a = q.top();
        q.pop();
        if(!q.empty()){
            b = q.top();
            q.pop();
            sum += a + b;
            q.push(a + b);
        }
        else break;
    }
    return sum;
}

下面欣赏zz代码：

#include<bits/stdc++.h>
using namespace std;
typedef struct{
    char c;
    int f;
}P;
typedef struct{
    char c, code[70];
    int len, key;
}Q;
P p[70];
Q q[70];
int Len[300];
map <int, int> vis[1010];
bool cmp1(P x, P y){
    if(x.f == y.f){
        return Len[x.c] < Len[y.c];
    }
    else return x.f > y.f;
}
bool cmp2(Q x, Q y){
    return x.len < y.len;
}
int main()
{
    int N, M;
    scanf("%d", &N); getchar();
    for(int i = 0; i < N; i++){
        scanf("%c %d", &p[i].c, &p[i].f); getchar();
    }
    scanf("%d", &M); getchar();
    while(M--){
        for(int i = 0; i < N; i++){
            scanf("%c%s", &q[i].c, q[i].code); getchar();
            Len[q[i].c] = q[i].len = strlen(q[i].code);//
        }
        sort(q, q + N, cmp2);
        sort(p, p + N, cmp1);//排序考虑长度 
        int cnt, l = 2, flag = 1, last = 0, Maxj = 0;
        //判断是否最优长度
        /*
        !!!WAWAWA 最优长度应指编码总长度， 与用Huffman算法算出来的编码相比较才对 
        下面欣赏zz代码 
        */
        for(int i = 0; i < N; ){
            cnt = 0;
            if(q[i].len - l > 0){flag = 0; break;}
            while(q[i].len == l && i < N){//同一深度计数 
                for(int j = 0; j < N; j++)
                    if(p[j].c == q[i].c) {Maxj = max(Maxj, j); break;}
                i++; cnt++;
            }
            //判断是否最优编码，更新深度 
            if(Maxj - last + 1== cnt) {last = Maxj + 1; cnt = 0;}
            else { flag = 0; break;}
            l++;
        }
        if(!flag) {printf("No\n");  continue;}
        //判断是否前缀编码 
        for(int i = 0; i < N; i++){
            int sum = 0;
            for(int j = 0; j < q[i].len; j++){
                sum = sum * 3 + (q[i].code[j] - '0' + 1);
                if(vis[M][sum]) {flag = 0; break;}
            }
            if(!flag) break;
            vis[M][sum] = 1;
        } 
        if(!flag) printf("No\n");
        else printf("Yes\n");
    }
    return 0;
}

优先队列真是贼jb好用
有空贴上手写的优先队列
不 没有空