1. 程式人生 > >基礎練習 Huffuman樹

基礎練習 Huffuman樹

問題描述

  Huffman樹在編碼中有著廣泛的應用。在這裡,我們只關心Huffman樹的構造過程。   給出一列數{pi}={p0, p1, …, pn-1},用這列數構造Huffman樹的過程如下:   1. 找到{pi}中最小的兩個數,設為papb,將papb從{pi}中刪除掉,然後將它們的和加入到{pi}中。這個過程的費用記為pa + pb。   2. 重複步驟1,直到{pi}中只剩下一個數。   在上面的操作過程中,把所有的費用相加,就得到了構造Huffman樹的總費用。   本題任務:對於給定的一個數列,現在請你求出用該數列構造Huffman樹的總費用。   例如,對於數列{pi

}={5, 3, 8, 2, 9},Huffman樹的構造過程如下:   1. 找到{5, 3, 8, 2, 9}中最小的兩個數,分別是2和3,從{pi}中刪除它們並將和5加入,得到{5, 8, 9, 5},費用為5。   2. 找到{5, 8, 9, 5}中最小的兩個數,分別是5和5,從{pi}中刪除它們並將和10加入,得到{8, 9, 10},費用為10。   3. 找到{8, 9, 10}中最小的兩個數,分別是8和9,從{pi}中刪除它們並將和17加入,得到{10, 17},費用為17。   4. 找到{10, 17}中最小的兩個數,分別是10和17,從{pi}中刪除它們並將和27加入,得到{27},費用為27。   5. 現在,數列中只剩下一個數27,構造過程結束,總費用為5+10+17+27=59。

輸入格式

  輸入的第一行包含一個正整數nn<=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;
}