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号。


版权声明:本文为zschzk607706原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。