最优合并问题————贪心

Problem Description

给定k 个排好序的序列s1 , s2,……, sk , 用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

和之前那个摆石子的一样的思路，不过比他简单多了

#include 
#include 
#include 
#include 
using namespace std;

int main()
{
    int n, i, a = 0, k = 2;
    priority_queue < int > q1;
    priority_queue < int, vector, greater > q2;
    cin>>n;
    for(i = 0; i < n; i++)
    {
        cin>>a;
        q1.push(a);
        q2.push(a);
    }
    int max1 = 0, max2 = 0;
    while(q1.size() > 1)
    {
        max2 = 0;
        for(i = 0; i < k; i++)
        {
            max2 += q1.top();
            q1.pop();
        }
        max1 += max2 - 1;
        q1.push(max2);
    }
    int min1 = 0, min2 = 0;
    while(q2.size() > 1)
    {
        min2 = 0;
        for(i = 0; i < k; i++)
        {
            min2 += q2.top();
            q2.pop();
        }
        min1 += min2 - 1;
        q2.push(min2);
    }
    printf("%d %d\n", max1, min1);
    return 0;
}