最优合并问题 （贪心算法）

最优合并问题
Time Limit: 1000 ms Memory Limit: 65536 KiB

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
Hint
Source

#include <bits/stdc++.h>

using namespace std;

//找最大的合并次数，即每次合并最大的两个序列
int findMax(int a[], int k)
{
    priority_queue<int> q;
    for(int i = 0; i < k; i++)
        q.push(a[i]);
    int sum = 0;
    while(q.size() >= 2)
    {
        //每次取出最大的两个序列进行合并
        int x = q.top();
        q.pop();
        int y = q.top();
        q.pop();
        sum += x + y - 1;  //合并次数
        q.push(x+y);      //合并后的序列长度继续放入数组
    }
    return sum;
}

//找最小的合并次数，即每次合并最小的两个序列
int findMin(int a[], int k)
{
    priority_queue<int,vector<int>,greater<int> > q;
    for(int i = 0; i < k; i++)
        q.push(a[i]);
    int sum = 0;
    while(q.size() >= 2)
    {
        //每次取出最小的两个序列进行合并
        int x = q.top();
        q.pop();
        int y = q.top();
        q.pop();
        sum += x + y - 1;  //合并次数
        q.push(x+y);      //合并后的序列长度继续放入数组
    }
    return sum;
}

int main()
{
    int k,i;
    int a[1009];
    cin>>k;
    for(i = 0; i < k; i++)
        cin>>a[i];
    int maxx = findMax(a,k);
    int minn = findMin(a,k);
    cout<<maxx<<" "<<minn<<endl;
    return 0;
}