数组重新排列。都小于 a 左边的 a。 a 的右边都大于 a

Question

我想重新排列我的数组 A，以便列表中大于任意数字 a 的所有元素都位于 a 的右侧strong>，所有小于 a 的元素都将位于 a 的左侧。

例子

A = [4,5,1,2]
a = A[0] #4

Output:

[1,2,4,5]
or
[2,1,4,5]

我试图从左到右分别找到数字 >= a 和 a。一旦找到，我会切换这些数字。

Answer 1

这是一个 python 函数，可以为您提供预期的输出

# here A is list, and a is A[x]
def arrangeit(A, a):
    finalLst = [a]
    for i in A:
        if(i < a):
            finalLst = [i] + finalLst
        if(i > a):
            finalLst.append(i)

    return finalLst

A = [4,5,1,2]
a = A[0] #4

print(arrangeit(A, a))