4.3 最优合并问题

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

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

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

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

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

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

78 52
作者
陈晓梅
单位
广东外语外贸大学
代码长度限制
16 KB
时间限制
400 ms
内存限制
64 MB

#include<iostream>  
#include<deque>
#include<algorithm>
using namespace std; 
bool cmp(int a,int b){
  return a>=b;
}
int fac1(deque<int> data,int n){
  int a;
  int b;
  int max_count;
  max_count=0;
  while(data.size()>1){
    sort(data.begin(),data.end(),cmp);
    a=data.front();
    data.pop_front();
    b=data.front();
    data.pop_front();
    max_count=max_count+a+b-1;
    data.push_back(a+b);
}
  return max_count;
}
int fac2(deque<int> data,int n){
  int a;
  int b;
  int min_count;
  min_count=0;
  while(data.size()>1){
    sort(data.begin(),data.end());
    a=data.front();
    data.pop_front();
    b=data.front();
    data.pop_front();
    min_count=min_count+a+b-1;
    data.push_back(a+b);
}
  return min_count;
}
int main(){
  int n;
  int a;
  deque <int> data;
  deque <int>:: iterator i;
   cin>>n;
   while(cin>>a){
      data.push_back(a);
}
  cout<<fac1(data,n)<<" "<<fac2(data,n)<<endl;
     return 0;
}