算法实现题 4-2 最优合并问题
给定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
贪心算法能够做到局部最优解,在本题中:
(1)如果你想要的是最多比较次数的合并方案,那么就要每一次合并的时候都是该序列中最长的那两个。
(2)如果你想要的是最少比较次数的合并方案,那么就要每一次合并的时候都是该序列中最短的那两个。
所采用的2 路合并算法合并2 个长度分别为m和n的序列需要m + n -1次比较(题目已给出),把这个序列合并在一起之后在接着计算,即可得出最优值。
所以,一定要先对序列进行排序。
例: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;
bool cmp(int a,int b) {
return a>b;
}
int MaxMerge(int a[],int k) {
int max=0;
for(int i=0; i<k-1; i++) {
sort(a+i,a+k,cmp);//降序排列 只对后面还没有操作的元素进行重新排序
a[i+1]=a[i]+a[i+1];
max=max+a[i+1]-1;
}
return max;
}
int MinMerge(int b[],int k) {
int min=0;
for(int i=0; i<k-1; i++) {
sort(b+i,b+k);//升序排列
b[i+1]=b[i]+b[i+1];
min=min+b[i+1]-1;
}
return min;
}
int main() {
int k;
cin>>k;
int a[k],b[k];
int x;
for(int i=0; i<k; i++) {
cin>>x;
a[i]=x;
b[i]=x;
}
cout<<MaxMerge(a,k)<<" "<<MinMerge(b,k)<<endl;
return 0;
}
/*
4
5 12 11 2
78 52
*/