给定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
例:
4 5 12 11 2
排序2 4 5 11 12
贪心策略:
- 每次选最小的序列合并得到最少比较次数;
- 每次选最大的序列合并得到最多比较次数。
- 2 个长度分别为m和n的序列需要m + n -1次比较
- 最多比较次数=(12+11-1)+ (12+11+5-1 )+ (12+11+5 +4 -1)+ (12+11+5+4+2-1 )
- 最少比较次数=2+4-1+5+6-1+11+11-1((((2+4-1)+5-1)+11-1)+12-1)
#include <bits/stdc++.h>
using namespace std;
priority_queue<int>q;//降序
priority_queue<int,vector<int>,greater<int> >p;//升序
int main()
{
int n;
scanf("%d",&n);
for(int i=0;i<n;i++)
{
int x;
scanf("%d",&x);
q.push(x);
p.push(x);
}
int maxx=0;
int minn=0;
while(q.size()>1)每次取出最大的两个序列进行合并
{
int x=q.top();
q.pop();
int y=q.top();
q.pop();
maxx+=x+y-1;// //合并次数
q.push(x+y);合并后的序列长度继续放入数组
}
while(p.size()>1)每次取出最小的两个序列进行合并
{
int x=p.top();
p.pop();
int y=p.top();
p.pop();
minn+=x+y-1;// //合并次数
p.push(x+y);合并后的序列长度继续放入数组
}
printf("%d %d\n",maxx,minn);
return 0;
}