python中的插入排序，列表末尾有负数

Question

我使用排序算法对数字进行排序，例如正数，然后是负数（就地交换，您可以使用变量，但不使用任何容器变量。假设我们没有更多的内存空间对于额外的容器）。所以先排序正数，再排序负数。

我做到了，但我需要另一个 for 循环来对负数进行排序并将它们插入到列表的末尾。

我的问题：有没有办法修改第一个循环以达到相同的结果？ （即：仅使用 1 个for 循环）

例如：

x = [9, -3, 2, -7, -4, -1, 5, 4]

输出应该是：

[2, 4, 5, 9, -7, -4, -3, -1]

这是我的解决方案（但我使用了 2 个 for 循环，我只想使用 1 个）：

def insertion_sort(data):
    for i in range(1, len(data)):
        e = i
        while e > 0 and data[e - 1] > data[e]:
            data[e - 1], data[e] = data[e], data[e - 1]
            e -= 1
    for i in range(1, len(data)):
        e = i
        while e > 0 and data[e - 1] < 0:
            if data[e] < 0:
                if data[e - 1] < data[e]:
                    data[e - 1], data[e] = data[e], data[e - 1]
            data[e - 1], data[e] = data[e], data[e - 1]
            e -= 1
    print(data)

Answer 1

你可以定义一个函数is_greater如下：

def is_greater(x, y):
    if (x < 0 and y < 0) or (x >= 0 and y >= 0):
        return x > y
    return x < 0

然后你可以在你的第一个循环中使用它而不是普通的 int 比较：

def insertion_sort(data):
    for i in range(1, len(data)):
        e = i
        while e > 0 and is_greater(data[e - 1], data[e]):
            data[e - 1], data[e] = data[e], data[e - 1]
            e -= 1
    print(data)

insertion_sort(d)

输出

[2, 4, 5, 9, -7, -4, -3, -1]

Answer 2

您只需要在数字开始为正数时找到枢轴值

def insertion_sort(data):

    pivot = -1
    for i in range(1, len(data)):
        e = i
        while e > 0 and data[e - 1] > data[e]:
            data[e - 1], data[e] = data[e], data[e - 1]
            if data[e] >= 0 and data[e-1] < 0:
                pivot = e
            e -= 1

    data[-pivot:], data[:-pivot] = data[:pivot], data[pivot:]
    print(data)