最优合并问题

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

code:

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

bool cmp(int a, int b)
{
    return a>b;
}

int main()
{
    int k;
    int a[1010], b[1010];
    int minval = 0, maxval = 0;
    cin>>k;
    for(int i = 0;i<k;i++)
    {
        cin>>a[i];
        b[i] = a[i];
    }
    sort(a, a+k);
    sort(b, b+k, cmp);
    for(int i = 0;i<k-1;i++)
    {
        a[i+1] = a[i]+a[i+1];
        minval += a[i+1];
        sort(a+i+1, a+k);

        b[i+1] = b[i]+b[i+1];
        maxval += b[i+1];
        sort(b+i+1, b+k, cmp);
    }
    cout<<maxval-k+1<<' '<<minval-k+1<<endl;
}