1035 插入与归并Java实现

1035 插入与归并

根据维基百科的定义：
插入排序是迭代算法，逐一获得输入数据，逐步产生有序的输出序列。每步迭代中，算法从输入序列中取出一元素，将之插入有序序列中正确的位置。如此迭代直到全部元素有序。
归并排序进行如下迭代操作：首先将原始序列看成 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

代码实现：

import java.util.Scanner;
import java.util.Arrays;
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int[] a = new int[n];//原始的
        int[] b = new int[n];//部分排序过的
        for(int i = 0;i<n;i++)
            a[i] = in.nextInt();
        for(int i = 0;i<n;i++)
            b[i] = in.nextInt();
        String type = "Insertion Sort";
        int index = 0;
        for(int i = 0;i<n-1;i++){
            if(b[i]>b[i+1]){
                index = i+1;//从i+1这里开始无序
                break;
            }
        }
        for(int i = index;i<n;i++){
            if(a[i]!=b[i]){
                type = "Merge Sort";//如果是插入排序，从index开两个数组的顺序是一样的
                break;
            }
        }
        if(type.equals("Insertion Sort")){
            int num = b[index];
            for(int j = index;j>0;j--){
                if(b[j]<b[j-1]){
                    b[j] = b[j-1];
                    b[j-1] = num;
                }
            }
        }else{
            //如果是归并排序 index的值表示这么多的规模已经排序完，
            //比如index==2 说明规模为2的已经排序完 下一次排序规模为4
            index = 2*index;
            for(int i = 0;i<n;i+=index){
                int next = i+index>n?n:i+index;
                Arrays.sort(b,i,next);
            }
        }
        System.out.println(type);
        for(int i = 0;i<n-1;i++)
            System.out.print(b[i]+" ");
        System.out.print(b[n-1]);
    }
}

运行结果：