最优合并问题【贪心算法】

• Problem Description

给定k个排好序的序列$s_1$， $s_2$，……，$s_k$， 用2 路合并算法将这k 个序列合并成一个序列。假设所采用的2 路合并算法合并2 个长度分别为m和n的序列需要m + n - 1次比较。试设计一个算法确定合并这个序列的最优合并顺序，使所需的总比较次数最少。
为了进行比较，还需要确定合并这个序列的最差合并顺序，使所需的总比较次数最多。
对于给定的k个待合并序列，计算最多比较次数和最少比较次数合并方案。

Input
输入数据的第一行有1 个正整数k（k≤1000），表示有k个待合并序列。接下来的1 行中，有k个正整数，表示k个待合并序列的长度。
Output
输出两个整数，中间用空格隔开，表示计算出的最多比较次数和最少比较次数。

Sample Input

4
5 12 11 2

Sample Output

78 52

• 解题思路

（1）该题与多元Huffman编码问题类似，可以用优先队列解答。
（2）2 路合并算法就是每次合并两个序列。可以证明：对递增序列可以求得最少比较次数，对递减序列可以求得最多比较次数。
（3）与多元Huffman编码问题稍微不同的地方：根据题意，每次合并完后，要将得到的sum值先减去1后，再加入到$su{m}_1$或$su{m}_2$中去。

• 代码实现

#include <bits/stdc++.h>
using namespace std;

int main()
{
    int k,x;
    long long sum1=0,sum2=0;   //定义最少比较次数sum1和最多比较次数sum2
    priority_queue<int, vector<int>, greater<int> > q1;
    priority_queue<int, vector<int>, less<int> > q2;
    cin>>k;
    for(int i=0;i<k;i++)
    {
        cin>>x;
        q1.push(x);
        q2.push(x);
    }
    while(q1.size()>1)
    {
        long long sum=0;
        for(int i=0;i<2;i++)
        {
            sum+=q1.top();
            q1.pop();
        }
        sum1+=sum-1;   //每合并两个序列，需要比较 m+n-1 次
        q1.push(sum);
    }
    while(q2.size()>1)
    {
        long long sum=0;
        for(int i=0;i<2;i++)
        {
            sum+=q2.top();
            q2.pop();
        }
        sum2+=sum-1;
        q2.push(sum);
    }
    cout<<sum2<<" "<<sum1<<endl;
    return 0;
}

• 运行结果