基礎練習 Huffuman樹
阿新 • • 發佈:2018-12-13
問題描述
Huffman樹在編碼中有著廣泛的應用。在這裡,我們只關心Huffman樹的構造過程。 給出一列數{pi}={p0, p1, …, pn-1},用這列數構造Huffman樹的過程如下: 1. 找到{pi}中最小的兩個數,設為pa和pb,將pa和pb從{pi}中刪除掉,然後將它們的和加入到{pi}中。這個過程的費用記為pa + pb。 2. 重複步驟1,直到{pi}中只剩下一個數。 在上面的操作過程中,把所有的費用相加,就得到了構造Huffman樹的總費用。 本題任務:對於給定的一個數列,現在請你求出用該數列構造Huffman樹的總費用。 例如,對於數列{pi
輸入格式
輸入的第一行包含一個正整數n(n<=100)。 接下來是n個正整數,表示p0, p1, …, pn-1,每個數不超過1000。
輸出格式
輸出用這些數構造Huffman樹的總費用。
樣例輸入
5 5 3 8 2 9
樣例輸出
59
很簡單的一道題,直接按要求來就好
#include<iostream> #include<stdio.h> #include<cstring> #include<algorithm> using namespace std; bool cmp(int a,int b){ return a<b;} int main(void){ int n,count=0; int p[105]; scanf("%d",&n); for(int i=0;i<n;i++){ scanf("%d",&p[i]); count++; } int k=0,sum=0,sum2=0; while(count>=2){ //陣列內元素多於兩個時 sort(p+k,p+k+count,cmp); //已經加過的第一個元素捨棄,從第二個開始排序 sum=p[k]+p[k+1]; //此次排位置前兩個的元素求和 k++; p[k]=sum; //已經加過的第二個元素的位置上放置兩個元素的和 count--; sum2+=sum; } printf("%d",sum2); return 0; }