最优合并问题

题目来源：王晓东《算法设计与分析》

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

输入格式:
第一行有 1 个正整数k，表示有 k个待合并序列。 第二行有 k个正整数，表示 k个待合并序列的长度。

输出格式:
输出最多比较次数和最少比较次数。

输入样例:
在这里给出一组输入。例如：

4
5 12 11 2
输出样例:
在这里给出相应的输出。例如：

78 52

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
int n;
int cmp(int a,int b)
{
	return a>b;
}
int better_sort(int a[],int m)
{
	int b[n];
	int sum=0;
	for(int i=0;i<n;i++)
		b[i]=a[i];
	while(m>1)
	{
		sort(b,b+m);
		b[0]=b[0]+b[1];
		sum+=b[0];
		for(int i=1;i<m-1;i++)
			b[i]=b[i+1];
		m--;
	}
	return sum-n+1;
}
int worse_sort(int a[],int m)
{
	int b[n];
	int sum=0;
	for(int i=0;i<n;i++)
		b[i]=a[i];
	while(m>1)
	{
		sort(b,b+m,cmp);
		b[0]=b[0]+b[1];
		sum+=b[0];
		for(int i=1;i<m-1;i++)
			b[i]=b[i+1];
		m--;
	}
	return sum-n+1;
}
int main()
{
	//int a[100];
	cin>>n;
    int a[n+1];
	for(int i=0;i<n;i++)
		cin>>a[i];
	cout<<worse_sort(a,n)<<" "<<better_sort(a,n)<<endl;
	return 0;
 }