Python【走出棋盘】

要求：

某个人进入如下一个棋盘中，要求从左上角开始走，
    最后从右下角出来（要求只能前进，不能后退），
    问题：共有多少种走法？
    
    0 0 0 0 0 0 0 0

    0 0 0 0 0 0 0 0

    0 0 0 0 0 0 0 0

    0 0 0 0 0 0 0 0

    0 0 0 0 0 0 0 0

代码如下：

n = int(input("请输入数组的长度："))
arr = list(map(int, input("输入n个数组：").split()))# 将第一个元素作为枢纽
pivot = arr[0]# 划分数组
less_than_pivot = [num for num in arr if num < pivot]
equal_to_pivot = [num for num in arr if num == pivot]
greater_than_pivot = [num for num in arr if num > pivot]# 输出划分后的结果
result = less_than_pivot + equal_to_pivot + greater_than_pivot
print(' '.join(map(str, result)))