剑指 Offer 62. 圆圈中最后剩下的数字

题目描述

https://leetcode-cn.com/problems/yuan-quan-zhong-zui-hou-sheng-xia-de-shu-zi-lcof
0,1,···,n-1这n个数字排成一个圆圈，从数字0开始，每次从这个圆圈里删除第m个数字（删除后从下一个数字开始计数）。求出这个圆圈里剩下的最后一个数字。

例如，0、1、2、3、4这5个数字组成一个圆圈，从数字0开始每次删除第3个数字，则删除的前4个数字依次是2、0、4、1，因此最后剩下的数字是3。

示例 1：

输入: n = 5, m = 3
输出: 3
示例 2：

输入: n = 10, m = 17
输出: 2

限制：

1 <= n <= 10^5
1 <= m <= 10^6

解题思路

这个也叫约瑟夫环。
关键点在每次删除的pos都是(pos + m - 1) % len(nums)。删除的方法为pop(pos)
一个小技巧：python生成0、1、2、3、4这样的list的方法为nums = list(range(5))。

解题代码

class Solution:
    def lastRemaining(self, n: int, m: int) -> int:
        nums = list(range(n))
        pos = 0
        while len(nums) > 1:
            pos = (pos + m - 1) % len(nums)
            nums.pop(pos)
        return nums[0]

执行结果

执行结果：通过
执行用时：2168 ms, 在所有 Python3 提交中击败了8.40%的用户
内存消耗：18.3 MB, 在所有 Python3 提交中击败了20.97%的用户