乙级(Basic Level) 1025 插入与归并

题目描述

根据维基百科的定义：



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

的位置。如此迭代直到全部元素有序。



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

下1个有序的序列。



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

输入描述:

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

空格分隔。

输出描述:

首先在第1行中输出“Insertion Sort”表示插入排序、或“Merge Sort”表示归并排序；然后在第2行中输出用该排序算法再迭代一轮的结果序列。题目保证每组测试

的结果是唯一的。数字间以空格分隔，且行末不得有多余空格。

输入例子:

10

3 1 2 8 7 5 9 4 6 0

1 2 3 7 8 5 9 4 6 0

输出例子:

Insertion Sort

1 2 3 5 7 8 9 4 6 0

from copy import deepcopy
def insert(lst,a):
    yes  = False
    for i in range(1,len(lst)):
        for j in range(i):
            if lst[j]>lst[i]:
                m = lst[i]
                for k in range(i,j,-1):
                    lst[k] = lst[k-1]
                lst[j] = m
                break
        if yes and lst!=a:
            break
        if lst==a:
            yes = True
    return yes,lst

def merge(lst,a):
    yes = False
    k = 2
    while True:
        for i in range(len(lst)//k+1):
            m = min(len(lst),(i+1)*k)
            m2 = min(len(lst),i*k+k//2)
            #print('i',i,k,m2)
            for j in range(i*k,m2):
                #print('j',j,end=' ')
                for k2 in range(m2,m):
                    #print('k2',k,end=' ')
                    if lst[j] > lst[k2]:
                        tmp = lst[j]
                        lst[j] = lst[k2]
                        lst[k2] = tmp
                        m3 = lst[m2]
                        if lst[m-1]<m3:
                            for k3 in range(m2+1,m):
                                lst[k3-1]=lst[k3]
                            lst[m-1] = m3
                            break
                        for k3 in range(m2+1,m):
                            if lst[k3]>m3:
                                for k4 in range(m2+1,k3):
                                    lst[k4-1] = lst[k4]
                                lst[k3-1] = m3
                                break
                    else:
                        break
        k *= 2
        if yes:
            break
        if lst==a:
            yes = True
    return lst

num = int(input())
a = input().split()
a = [int(i) for i in a]
b = input().split()
b = [int(i) for i in b]
c = deepcopy(a)
yes,c = insert(c,b)
if yes:
    print("Insertion Sort")
    for i in c:
        print(i,end=' ')
    #print(str(c)[1:-1])
else:
    print("Merge Sort")
    c = merge(a,b)
    for i in c:
        print(i,end=' ')
    #print(merge(a,b))