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 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;
}