多元Huffman编码问题

多元Huffman编码问题

时间： 1ms        内存：64M

描述：

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

对于给定n堆石子,计算合并成一堆的最大总费用和最小总费用。

输入：

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

输出：

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

示例输入：

7 3
45 13 12 16 9 5 22

示例输出：

593 199

提示：

参考答案（内存最优[3908]）：

#include<iostream>
#include<cstdio>
#include<list>
#include<queue>
#include<vector>
#include<algorithm>
#include<set>
using namespace std;
int n, k, total, maxVal, minVal, temp, num, p[100005];
multiset<int> mySet;

void solve()
{ 
    while (mySet.size() != 1)
    {
        temp = 0;
        for (int i=0; i<k; i++)
        {
            temp += *mySet.begin();
            mySet.erase(mySet.begin());
        }
        mySet.insert(temp);  
        minVal += temp;
    }
}

int main()
{
    //while (scanf("%d %d", &n, &k) != EOF)
    scanf("%d %d", &n, &k);
    {
          total = temp = minVal = maxVal = 0;
          mySet.clear();
          for (int i=0; i<n; i++)
          {
              scanf("%d", &num);
              p[i] = num;
              mySet.insert(p[i]);
              total += num;
          }
          if (k > n) k = n;
          temp = 0;
          sort(p, p+n);
          for (int i=n-1; i>=0; i--)
          {
              temp += p[i];
              if (i != n-1) maxVal += temp;
          }  
          while (mySet.size() % (k - 1) != 1 && k != 2) mySet.insert(0);
          solve();
          printf("%d %d\n", maxVal, minVal);                    
    }  
    return 0;
}

参考答案（时间最优[152]）：

#include<iostream>
#include<cstdio>
#include<list>
#include<queue>
#include<vector>
#include<algorithm>
#include<set>
using namespace std;
int n, k, total, maxVal, minVal, temp, num, p[100005];
multiset<int> mySet;

void solve()
{ 
    while (mySet.size() != 1)
    {
        temp = 0;
        for (int i=0; i<k; i++)
        {
            temp += *mySet.begin();
            mySet.erase(mySet.begin());
        }
        mySet.insert(temp);  
        minVal += temp;
    }
}

int main()
{
    //while (scanf("%d %d", &n, &k) != EOF)
    scanf("%d %d", &n, &k);
    {
          total = temp = minVal = maxVal = 0;
          mySet.clear();
          for (int i=0; i<n; i++)
          {
              scanf("%d", &num);
              p[i] = num;
              mySet.insert(p[i]);
              total += num;
          }
          if (k > n) k = n;
          temp = 0;
          sort(p, p+n);
          for (int i=n-1; i>=0; i--)
          {
              temp += p[i];
              if (i != n-1) maxVal += temp;
          }  
          while (mySet.size() % (k - 1) != 1 && k != 2) mySet.insert(0);
          solve();
          printf("%d %d\n", maxVal, minVal);                    
    }  
    return 0;
}

题目和答案均来自于互联网，仅供参考，如有问题请联系管理员修改或删除。