貪心算法 Problem D 1003 哈夫曼編碼

Problem D  Problem ID：1003

簡單題意：給出一組字符串，求出普通編碼將佔用的位數和哈夫曼編碼所用的位數，以及普通編碼與哈夫曼編碼比率（普通編碼長度除以哈夫曼編碼長度）。

解題思路形成過程：哈夫曼編碼所佔用的位數正好等於哈夫曼樹各個非根節點的權值之和。因此，將字符儲存好後按每個字母的數量將其進行排序，從小到大進行遍歷相加進行哈夫曼樹的組建，與此同時對哈夫曼樹各個非根節點的權值進行累加即可得結果。

感想：①注意題目中蘊含的數學關係，數學關係理清則很可能想出事半功倍的算法。

   ②注意邊界條件，看邊界值是否符合要求。

代碼：

#include<iostream>
#include<stdio.h>
#include<cstring>
#include<string.h>
#include<algorithm>
using namespace std;
int a[30];
char s[2000001];
bool cmp(int a,int b){
    return a>b;
}
int main()
{
    while(cin>>s){
        if(strcmp(s,"END")==0)
            return 0;
        int num,cnt=0;
        memset(a,0,sizeof(a));
        int len=strlen(s);
        for(int i=0;i<len;++i){
            if(s[i]=='_')
                ++a[0];
            else
                ++a[s[i]-'A'+1];
        }
        sort(a,a+27,cmp);
        for(int i=0;;++i)
            if(a[i]==0){
                num=i;
                break;
            }
        for(int i=num-1;i>0;--i){
            a[i-1]+=a[i];
            cnt+=a[i-1];
            sort(a,a+num,cmp);
        }
        if(num==1)//注意邊界條件！
            cnt=a[0];
        printf("%d %d %.1f\n",8*len,cnt,8.0*len/cnt);//注意保留一位小數的寫法。
    }
}