面试算法题-排序(某外包岗)

"""
对于两个有序数组:
nums1 = [1, 3, 5, 8]
nums2 = [1, 2, 6]

m = nums1的长度
n = nums2的长度
可将nums1的长度视为 = m + n, 即
nums1 = [1, 3, 5, 8, 0, 0, 0]

将两个列表合并后排序:
nums1 = [1, 1, 2, 3, 5, 6, 8]
"""

可惜面试时没能想出好的解法，只是用归并算法的原理新建了一个列表做的。

面试结束后思考了一下，可以从末尾开始比较大小。

import random
# 随机生成nums1, nums2并排序
nums1 = [random.randint(1, 100) for i in range(5)]
nums2 = [random.randint(1, 100) for i in range(4)]
nums1 = sorted(nums1)
nums2 = sorted(nums2)
# nums1后补0
nums1.extend([0 for i in range(len(nums2))])

n = len(nums2)
m = len(nums1) - n

# n_指向nums2的末尾, m_指向nums1最后一个非0数字, index_指向nums1末尾
n_ = n - 1
m_ = m - 1
index_ = m + n - 1

# 当两个数组指针有效时
while n_ >= 0 and m_ >= 0:
    # m_数大则与index_位置互换, n_大则赋值到index_位置
    if nums1[m_] > nums2[n_]:
        nums1[m_], nums1[index_] = nums1[index_], nums1[m_]
        m_ -= 1
    else:
        nums1[index_] = nums2[n_]
        n_ -= 1
    # nums1的末尾指针减一
    index_ -= 1
# 若nums1遍历完, 则将nums2只剩余数字赋值到index_
while n_ >= 0:
    nums1[n_] = nums2[n_]
    n_ -= 1