试题 基础练习 Huffuman树

题目

问题描述
　　Huffman树在编码中有着广泛的应用。在这里，我们只关心Huffman树的构造过程。
　　给出一列数{pi}={p0, p1, …, pn-1}，用这列数构造Huffman树的过程如下：
　　1. 找到{pi}中最小的两个数，设为pa和pb，将pa和pb从{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。
输入格式
　　输入的第一行包含一个正整数n（n<=100）。
　　接下来是n个正整数，表示p0, p1, …, pn-1，每个数不超过1000。
输出格式
　　输出用这些数构造Huffman树的总费用。
样例输入
5
5 3 8 2 9
样例输出
59

思路

• 方法一 ：优先级队列
          使用优先级队列反复执行弹出2个元素，插入两个元素的和操作即可。

• 方法二 ： 哨兵 + 数组的形式

        在队列中标记出最小和次小，将其交换到末尾位置实现假删除，然后将其和再插入到末尾即可。

实现代码

• 方法一 :

#include <bits/stdc++.h>

using namespace std;

// 比较器
struct cmp
{
    bool operator()(const int &lhs, const int &rhs)
    {
        return lhs >= rhs;
    }
};

int main()
{
    priority_queue<int, vector<int>, cmp> Q;
    
    // 输入
    int n, t; cin >> n;
    for(int i = 0; i < n; i++)
    { 
        cin >> t;
        Q.push(t);
    }

    // 构造haffman
    int ans = 0;
    while(Q.size() > 1)
    {
        
        int a = Q.top(); Q.pop();
        int b = Q.top(); Q.pop();
        
        int c = a + b; ans += c;
        cout << c << " = " << a << " + " << b << endl;
        Q.push(c);
    }
    cout << ans << endl;
    return 0;
}

• 方法二 :

#include <bits/stdc++.h>

using namespace std;

int a[100];
int k = 0;          // 总是指向待插入的位置


// 计算费用，并进行删除和插入操作
int ope()
{
        // 一次循环确定两个最小的数字的下标
    int first = INT_MAX, second = INT_MAX;
    int f_indx = -1, s_indx = -1;
    for(int i = 0; i < k; i++)
    {
        if(a[i] <= first)
        {
            // 将当前最小标记为次小
            second = first;
            s_indx = f_indx; 

            // 重新标记最小
            first = a[i];
            f_indx = i;
        }
        else if(a[i] < second)
        {
            second = a[i];
            s_indx = i;
        }
    }
    // 删除两个元素
    if(s_indx + 1 == k)
    {
        swap(a[s_indx], a[--k]);
        swap(a[f_indx], a[--k]);
    }
    else
    {
        swap(a[f_indx], a[--k]);
        swap(a[s_indx], a[--k]);
    }
    
    // 插入元素
    a[k++] = first + second;

 
    // 返回计算结果
    int ans = first + second;
    return ans;
}

int main()
{
   // 输入
    int n; cin >> n;
    for(int i = 0; i < n; i++)
    {
        cin >> a[i];
        k++;
    }

    // 计算费用
    int ans = 0;
    while(k > 1)
    {
        ans += ope();
    }

    cout << ans << endl;
    return 0;
}