问题描述:
有一个数组a[N]顺序存放0~N-1,要求每隔两个数删掉一个数,到末尾时循环至开头继续进行,求最后一个被删掉的数的原始下标位置。以8个数(N=7)为例:{0,1,2,3,4,5,6,7},0->1->2(删除)->3->4->5(删除)->6->7->0(删除),如此循环直到最后一个数被删除。
输入:
每组数据为一行一个整数n(小于等于1000),为数组成员数,如果大于1000,则对a[999]进行计算。
输出:
一行输出最后一个被删掉的数的原始下标位置
题目来源于牛客网
问题分析
这是一个典型的约瑟夫热土豆问题,使用队列进行模拟可以解决。具体的思路为:对于队列中的每个元素,给予0-2的动态编号;进行遍历队列,当前元素的编号如果小于2时,将该元素弹出并加入到队尾,否则将该元素弹出并删除。直至遍历到队列为空,记录最后遍历的数字。
并注意num的取值范围为<1000
代码
import sys
from queue import deque
def delNum(num):
num = min(num,1000)
que = deque([x for x in range(num)])
cnt = res = 0
while que:# que需要循环到为空的时候,将最后一个数字记录下来
res = que.popleft()
if cnt < 2:
que.append(res)
cnt += 1
else:
cnt = 0
return res
if __name__ == "__main__":
for num in sys.stdin:
print(delNum(int(num)))