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

根据维基百科的定义：

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

归并排序进行如下迭代操作：首先将原始序列看成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<cstdio>
#include<algorithm>
using namespace std;

int main(){
	
	int n;
	scanf("%d",&n);

	int a[n];
	int b[n];
	
	for(int i=0 ;i<n ;i++){
		scanf("%d",&a[i]);
	}
	
	for(int i=0 ;i<n ;i++){
		scanf("%d",&b[i]);
	}
	

	int i,j;
	for(i=0 ;i<n-1 && b[i+1]>=b[i] ;i++); //i记录了第一个相同元素的位置 
	for(j=i+1 ;j<n && a[j]==b[j];j++);	//j记录了最后一个相同元素的位置+1 
	
	if(j==n){	//如果最后一个相同的位置是n-1 那么说明是插入排序 
		printf("Insertion Sort\n");
		sort(a,a+i+2);	

	}else{
		printf("Merge Sort\n");
	
		int k = 1;	//下一次进行排序的子序列长度是2*k 
        int flag = 1;
       	
       	//设置flag变量后，模拟归并过程。直到达到给出的状态，然后再进行一次归并，将结果输出即可。 
	    while(flag) {
            flag = 0;
            for (i = 0; i < n; i++) {
                if (a[i] != b[i])
                    flag = 1;
            }
            
			k = k * 2;
            for (i = 0; i < n / k; i++){
            	sort(a + i * k, a + (i + 1) * k);
			}
			sort(a + n / k * k, a + n);
        }
		
	}
	
	for(int i=0 ;i<n-1 ;i++){
		printf("%d ",a[i]);
	}
	printf("%d\n",a[n-1]);
	return 0;
}