C++：多元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

Hint

请注意数据范围是否可能爆 int。

Source

#include <iostream>
#include <set>
#include <algorithm>

using namespace std;

class cmp        //仿函数
{
public:
    bool operator()(int a, int b)
    {
        return a > b;
    }
};

long long int getmax(int n, multiset<int, cmp> p)
{
    long long int sum = 0;
    while (p.size() > 2)
    {
        int mark = 0;
        for (int i = 0; i < 2; i++)
        {
            mark += *p.begin();
            p.erase(p.begin());
        }
        sum += mark;
        p.insert(mark);
    }
    while (!p.empty())
    {
        sum += *p.begin();
        p.erase(p.begin());
    }
    return sum;
}

long long int getmin(int n, int k, multiset<int> q)
{
    long long int sum = 0;
    if (q.size() > 2)
    {
        int m = n;
        while (m > k)
        {
            m = m - k + 1;
        }

        for (int i = 0; i < k - m; i++)
        {
            q.insert(0);
        }
    }
    while (q.size() > k)
    {
        int mark = 0;
        for (int i = 0; i < k; i++)
        {
            mark += *q.begin();
            q.erase(q.begin());
        }
        sum += mark;
        q.insert(mark);
    }
    while (!q.empty())
    {
        sum += *q.begin();
        q.erase(q.begin());
    }
    return sum;
}
int main()
{
    int n, k;
    cin >> n >> k;
    multiset<int, cmp> p;
    multiset<int> q;
    for (int i = 0; i < n; i++)
    {
        int x;
        cin >> x;
        p.insert(x);
        q.insert(x);
    }
    cout << getmax(n, p) << " ";
    cout << getmin(n, k, q) << endl;
    system("pause");
}