题目描述:
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
思路1:
直接暴力求解,去模拟整个过程,知道只剩下一个数字:
class Solution:
def lastRemaining(self, n: int, m: int) -> int:
nums = [i for i in range(n)]
k = m % len(nums) - 1
while len(nums) != 1:
nums.pop(k)
k = (k+(m-1)) % len(nums)
return nums[0]
这样时间复杂度会比较高,因为while进行一次循环,pop还要遍历,时间复杂度为O(n^2),由于n和m都是10 ^5 以上,所以会超时。
思路2: 动态规划 | 递归 | 约瑟夫环问题 (经典问题)
class Solution:
def lastRemaining(self, n: int, m: int) -> int:
res = 0
for i in range(2, n+1):
res = (res+m)%i
return res
以上思路参考了优秀的题解,题解链接
动态规划 | 递归 | 约瑟夫环问题 (经典问题)
参考了题解
# Python 默认的递归深度不够,需要手动设置
sys.setrecursionlimit(100000)
def f(n, m):
if n == 0:
return 0
x = f(n - 1, m)
return (m + x) % n
class Solution:
def lastRemaining(self, n: int, m: int) -> int:
return f(n, m)
递归的方式相对的空间复杂度会比较高。
无论是递归还是动态规划思路是一样的。