检查数组是否可堆栈排序

堆栈排序是指对一个序列进行排序，使得每次将一个元素插入堆栈顶部后都是有序的状态。可以将其看做对一个未排序序列进行插入排序的过程。

对于一个数组而言，如果它可以被堆栈排序，则表示该数组可以被排成一个有序的序列。现在，我们需要编写一个函数来检查一个数组是否满足可堆栈排序的条件。

实现思路

我们可以使用一个栈来辅助实现堆栈排序。遍历数组的每一个元素，如果当前元素比栈顶元素大，则将当前元素入栈；否则，将栈顶元素弹出，并将栈中元素插入到对应的位置。

实现代码如下：

def is_stack_sortable(arr):
    stack = []
    for n in arr:
        while stack and stack[-1] > n:
            arr[stack.pop()] = n
        stack.append(len(arr) - 1)
    return arr == sorted(arr)

函数说明

该函数名为 is_stack_sortable，接受一个数组作为参数，返回布尔值。当数组可堆栈排序时，返回 True，否则返回 False。

使用示例

arr1 = [3, 2, 1]
arr2 = [1, 2, 3]
print(is_stack_sortable(arr1))  # True
print(is_stack_sortable(arr2))  # True

注意事项

该函数并不会改变原始数组，而是返回判断结果的布尔值。如果需要对原始数组进行排序，可以在函数外部调用 sorted 函数。