每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0…m-1报数….这样下去….直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!^_^)。请你试着想下,哪个小朋友会得到这份礼品呢?(注:小朋友的编号是从0到n-1)
思路一:用一个数组去模拟整个过程,数组动态变化的话就可以跳着走。
# -*- coding:utf-8 -*-
class Solution:
def LastRemaining_Solution(self, n, m):
# write code here
if n < 1 or m < 1:
return -1
l = [i for i in range(n)]
j = 0
while len(l) > 1:
j = (j + m - 1) % len(l)
l.remove(l[j])
return l[0]
思路二:找规律。
这是一个约瑟夫环问题,在这n个数字中,第一个被删除的数字是(m-1)%n,将该数记为k。剩下的数字组成了一个新的约瑟夫环,下面我们把编号做一下映射,变换之后就变成了初始问题的子问题,如果我们知道这个子问题的解x,并将解进行逆映射(x+k+1)%n,就可以得到初始问题的解。
# -*- coding:utf-8 -*-
class Solution:
def LastRemaining_Solution(self, n, m):
# write code here
if n < 1 or m < 1:
return -1
last = 0
for i in range(2, n + 1):
last = (last + m) % i
return last