PAT-B 1035. 插入与归并(25)

题目链接在此。

思路

本题需要模拟插入排序和归并排序的每一步，并且归并排序用非递归形式会方便一些。整体做法是：先进行插入排序，在执行的过程中与给定序列是否相等，如果相等，则再进行一次排序后输出相应信息就可以结束算法；如果不是插入排序，则进行归并排序，在执行过程中与给定序列比较是否相等，题目设定一定会相等，那么找到吻合的那一次排序，再进行一次排序后输出相应信息即可。

AC代码

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

int n;

//判断两个数组是否相等
bool isSame(int a[], int b[]){
int i;
bool flag = true;
for(i = 0; i < n; i++){
if(a[i] != b[i]) {
flag = false;
}
}
return flag;
}

//打印数组
void printArr(int z[]){
int i;
for(i = 0; i < n; i++){
printf("%d",z[i]);
if(i < n-1){
printf(" ");
}
}
printf("\n");
}

//插入排序
bool insertion(int a[], int b[]){ //对a进行插入排序，b是中间序列，用来比较
int i,j;
bool print = false; //是否可以打印的标志位，当测试发现的确是插入排序，print=true，那么下一步排序就打印序列
for(i = 1; i < n; i++){
int temp = a[i];  //待排序列第一个元素
j = i;
while(j > 0 && temp < a[j-1]){ //元素右移，直到temp < a[j-1]
a[j] = a[j-1];
j--;
}
a[j] = temp; //元素归位
if(print == true){ //如果可以打印
printf("Insertion Sort\n"); //打印语句信息
printArr(a); //打印数组
return true; //返回真
}
if(isSame(a,b)) { //如果a,b相等
print = true; //设可以打印位为真
}
}
return false; //每一趟排序都不能使得a,b相等，返回false，表示不是这种排序策略
}

//归并排序
//对数组a的间进行归并排序，b是用来比较的数组
bool merge(int a[],int<
b99a
/span> b[]){
bool print = false;
for(int step = 2; step/2 <=n ; step*=2){ //将数组中的元素分组
for(int i = 0; i < n; i+=step){ //对每组进行排序
int left = i, right = min(n-1,i+step-1); //left&right分别是每一组的左右边界
sort(a+left,a+right+1); //对一组中的元素排序，由于时间允许，用sort函数代替合并的partition函数，注意sort函数的第二个参数是右边界+1
}
if(print){ //如果可以打印
printf("Merge Sort\n"); //打印语句信息
printArr(a); //打印数组
return true; //返回真
}
if(isSame(a,b)){ //如果a,b数组相等
print = true;   //置可打印标志位为1
}
}
return false; //每一趟排序都不能使得a,b相等，返回false，表示不是这种排序策略
}

int main(){

scanf("%d",&n);
int a[105],b[105]; //保存输入的两个数组

int for_a[105]; //用来备份a数组，为了防止测试的第一种排序方案不符时需要用它测试第二种
for(int i = 0; i < n; i++){ //读取a数组
scanf("%d",&a[i]);
for_a[i] = a[i];
}
for(int i = 0; i < n; i++){ //读取b数组
scanf("%d",&b[i]);
}

bool flag = insertion(a,b); //先测试插入排序
if(flag == false){ //如果不是插入排序
merge(for_a,b); //那就是归并排序 ，用a数组的备份数组for_a传入
}

return 0;
}