Python趣味练习题（一） - 射击游戏

题目如下：

假设有10000人，依次排列，首位相连，成为一个圈，10000跟1是相连的。第一个向第二个开枪，依次进行。求最后，谁能活下来。

先来看看5个人情况

1号开枪射杀2号：

2号死亡，轮到3号开枪射杀4号：

 

以此类推5号射杀1号，3号射杀5号，所以最终存活的是3号。

可以把这个问题抽象成：

有一个N项的数组，删除第二项后把第一项排到数组最后一位，循环这个过程直到数组内只有一个元素，用到了numpy.delete和numpy.roll，实现代码如下：

import numpy as np
arr = np.arange(1,10001)    # 可以根据题目修改范围
while arr.shape[0] > 1:
    arr = np.roll(np.delete(arr,1),-1)
arr
# Output: array([3617])

 当有10000人时，最后存活的是3617号。