1035. 插入与归并(25)

1035. 插入与归并(25)

时间限制

200 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CHEN, Yue

根据维基百科的定义：

插入排序是迭代算法，逐一获得输入数据，逐步产生有序的输出序列。每步迭代中，算法从输入序列中取出一元素，将之插入有序序列中正确的位置。如此迭代直到全部元素有序。

归并排序进行如下迭代操作：首先将原始序列看成N个只包含1个元素的有序子序列，然后每次迭代归并两个相邻的有序子序列，直到最后只剩下1个有序的序列。

现给定原始序列和由某排序算法产生的中间序列，请你判断该算法究竟是哪种排序算法？

输入格式：

输入在第一行给出正整数N (<=100)；随后一行给出原始序列的N个整数；最后一行给出由某排序算法产生的中间序列。这里假设排序的目标序列是升序。数字间以空格分隔。

输出格式：

首先在第1行中输出“Insertion Sort”表示插入排序、或“Merge Sort”表示归并排序；然后在第2行中输出用该排序算法再迭代一轮的结果序列。题目保证每组测试的结果是唯一的。数字间以空格分隔，且行末不得有多余空格。

输入样例1：

10
3 1 2 8 7 5 9 4 6 0
1 2 3 7 8 5 9 4 6 0

输出样例1：

Insertion Sort
1 2 3 5 7 8 9 4 6 0

输入样例2：

10
3 1 2 8 7 5 9 4 0 6
1 3 2 8 5 7 4 9 0 6

输出样例2：

Merge Sort
1 2 3 8 4 5 7 9 0 6

#include<stdio.h>
#include<algorithm>
using namespace std;

void Judge(int A[],int B[],int N,int &type,int &idx){
    //idx为直接插入已排序个数，或为归并排序间隔，type为排序类型，0为直接插入，1为归并
    int i;
    for(i=1;i<N;++i){   //寻找上升序列
        if(B[i]<B[i-1]){
            idx = i;
            break;
        }
    }
    for(;i<N;++i){
        if(A[i]!=B[i]){  //如果是直接插入排序，则后面的应一直相等，不等则为归并排序
            type = 1;   //归并排序类型为1
            return;
        }
    }
    type = 0;
}

void InsertionSort(int A[],int N,int idx){    //下一趟直接插入
    if(idx<N)
        sort(A,A+idx+1);
}

void MergeSort(int A[],int B[],int N){    //下一趟归并排序
    int d=1;
    while(1){
        bool flag=0;
        for(int i=0;i<N;++i){
            if(A[i]!=B[i]){
                flag = 1;
                break;
            }
        }
        d *= 2;
        for(int i=0;i<N/d;++i)
            sort(A+i*d,A+(i+1)*d);
        sort(A+(N/d)*d,A+N);
        if(flag==0) return;
    }
}

int main(){
    int N,A[100],B[100],type,idx;
    scanf("%d",&N);
    for(int i=0;i<N;++i)
        scanf("%d",&A[i]);
    for(int i=0;i<N;++i)
        scanf("%d",&B[i]);
    Judge(A,B,N,type,idx);  //判断是哪种排序并给出当前下标
    if(type==0){
        printf("Insertion Sort\n");
        InsertionSort(A,N,idx);   //继续插入
    }
    else{
        printf("Merge Sort\n");
        MergeSort(A,B,N);   //继续归并
    }
    for(int i=0;i<N-1;++i)  //输出下一趟排序结果
        printf("%d ",A[i]);
    printf("%d\n",A[N-1]);
    return 0;
}