题目:
每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0…m-1报数…这样下去…直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!_)。请你试着想下,哪个小朋友会得到这份礼品呢?(注:小朋友的编号是从0到n-1)
如果没有小朋友,请返回-1
思路:
用列表模拟环
代码一:
class Solution:
def LastRemaining_Solution(self, n, m):
if not n: return -1
arr = [i for i in range(n)]
i, count = 0, 0
while len(arr) > 1:
if i == m-1:
if i+count >= len(arr):
count = (i+count)%len(arr) - i
arr.pop(i+count)
count += m-1
i = 0
i += 1
return arr[0]
代码二:
class Solution:
def LastRemaining_Solution(self, n, m):
if not n: return -1
arr = [i for i in range(n)]
i, step, count = 0, 0, n
while count > 0:
i += 1
if i == len(arr): i = 0
if arr[i] == -1: continue
step += 1
if step == m-1:
arr[i] = -1
step = -1
count -= 1
return i