剑指offer孩子们的游戏(圆圈中最后剩下的数)

题目描述

每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0...m-1报数....这样下去....直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!^_^)。请你试着想下,哪个小朋友会得到这份礼品呢？(注：小朋友的编号是从0到n-1)

思路

1. 每次删除索引为(m-1+i)%len(n)的值
2. m-1是因为我们刚开始要删除第m个元素，列表从0位开始
3. m-1+i，+i的原因是我们要从当前删除值的地方(索引)开始
4. 直到只有一个值

# -*- coding:utf-8 -*-
class Solution:
    def LastRemaining_Solution(self, n, m):
        # write code here
        if not n or not m:
            return -1
        res=range(n)
        i=0
        while len(res)>1:
            i=(m-1+i)%len(res)
            res.pop(i)
        return res[0]

测试用例

if __name__=='__main__':
    s=Solution()
    n=5
    m=3
    print(s.LastRemaining_Solution(n,m))