多元Huffman编码问题

深藏阁楼爱情的钟 2022-02-15 04:02 200阅读 0赞

### 多元Huffman编码问题 ###

Time Limit: 1000 ms Memory Limit: 65536 KiB

[Submit][] [Statistic][]

Problem Description

在一个操场的四周摆放着n堆石子。现要将石子有次序地合并成一堆。规定每次至少选2 堆最多选k堆石子合并成新的一堆，合并的费用为新的一堆的石子数。试设计一个算法，计算出将n堆石子合并成一堆的最大总费用和最小总费用。  
对于给定n堆石子,计算合并成一堆的最大总费用和最小总费用。

Input

输入数据的第1 行有2 个正整数n和k（n≤100000，k≤10000），表示有n堆石子，每次至少选2 堆最多选k堆石子合并。第2 行有n个数（每个数均不超过 100），分别表示每堆石子的个数。

Output

将计算出的最大总费用和最小总费用输出，两个整数之间用空格分开。

Sample Input

7 3
    45 13 12 16 9 5 22

Sample Output

593 199

#include <bits/stdc++.h>
    
    using namespace std;
    
    int main()
    {
        priority_queue <int> queue1;
        priority_queue <int, vector <int>, greater <int> > queue2;
        int n, k;
        int x;
        long long sum1, sum2;
        scanf("%d%d", &n, &k);
        for(int i = 1; i <= n; i++)
        {
            scanf("%d", &x);
            queue1.push(x);
            queue2.push(x);
        }
        sum1 = 0;
        while(queue1.size() > 1)
        {
            int num1 = queue1.top();
            queue1.pop();
            int num2 = queue1.top();
            queue1.pop();
            queue1.push(num1 + num2);
            sum1 += num1 + num2;
        }
        while(queue2.size() % (k - 1) != 1) queue2.push(0);
    
        sum2 = 0;
        while(queue2.size() > 1)
        {
            int num2;
            int temp = 0;
            for(int i = 1; i <= k; i++)
            {
                num2 = queue2.top();
                queue2.pop();
                temp += num2;
            }
            queue2.push(temp);
            sum2 += temp;
        }
        printf("%lld %lld\n", sum1, sum2);
    }

[Submit]: https://acm.sdut.edu.cn/onlinejudge2/index.php/Home/Contest/contestsubmit/cid/2850/pid/1760.html
[Statistic]: https://acm.sdut.edu.cn/onlinejudge2/index.php/Home/Contest/conteststatus/cid/2850/pid/1760.html