一道作业题 变种插入排序

Description

---

　　设子数组A[0:k]和A[k+1:N-1]已排好序(0≤K≤N-1)。试设计一个合并这2个子数组为排好序的数组A[0:N-1]的算法。要求算法在最坏情况下所用的计算时间为O(N)，只用到O(1)的辅助空间。

思路

---

　　这道题难点在于辅助空间为O(1)，如果不是O(1)的话利用非比较排序能实现O(n)排序。

　　这道题突破口在于两个有序数组，前不了我了解到两个有序数组合并成一个有序数组可以使用插入排序进行归并，算法时间复杂度是O(n)，（合并排序中的merge方法），因为在将b[j]插入新数组时只需要和a[i]比较，如果b[j] > a[i]，由插入排序的性质可知b[j]比a[0..i-1]的任一元素都要大，所以直接放入a[i]后面即可，否则a[i]就放在b[j]之后，通过这样的办法就少了一层循环，达到O(n)。那么这道题也利用插入排序的思想试试吧。

　　首先用 i 指向第一个子数组的首位置，j指向第二个子数组的首位置，由于这道题没有O(n)的辅助空间，所以我们的想法需要变一变，维护一个串a[0..i]，对于序偶(a[0], a[1]), (a[1], a[2]), ..(a[i-1], a[i])都满足“<”关系。同时利用一个变量idx维护串a[idx..j]，其表示为小于a[i]的串。我们利用i、j遍历数组，当 a[i] <= a[j] 时，i指针后移；当a[i] > a[j]时，令inx = j，维护串a[inx..j] 并插入a[i-1]到a[i]之间。

　　图例如下：

　　　　

　　　　　　　　　来源于Google图片搜索，侵删

　　python代码如下，初试python，风格可能不太好：

# -*- coding: utf-8 -*-

def method(nums, k):
    i = 0
    j = k
    idx = -1
    left = len(nums)
    while i < j and j < left:
        while nums[i] <= nums[j]:
            i = i + 1
        idx = j
        while nums[i] > nums[j]:
            j = j + 1
            if j == left:
                break
        nums = nums[0:i] + nums[idx:j] + nums[i:idx] + nums[j:]
        i = i + j - idx
    return nums;



if __name__ == "__main__":
    a = [10, 30, 50, 60, 80, 20, 25, 55, 65, 75]
    k = 5
    a = method(a, k)
    print(a)