剑指 Offer 62. 圆圈中最后剩下的数字
题目描述:
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
解题思路:
可以看到n和m的值很大,不可暴力破解,本来表示超时很多次,o(╥﹏╥)o
看到大神用递推,膜拜!有兴趣的可以看一下,这里不多说了。。。
暴力破解代码:(超时,不提倡)
class Solution(object):
def lastRemaining(self, n, m):
"""
:type n: int
:type m: int
:rtype: int
"""
num, cnt = n, 1
list = [i for i in range(n)]
i = 0
while num > 1:
if cnt == m:
list.pop(i)
cnt = 1
num -= 1
if i == num :
i = 0
else:
i += 1
cnt += 1
if i >= num:
i = 0
return list[0]
动态规划代码:(提倡)
class Solution(object):
def lastRemaining(self, n, m):
"""
:type n: int
:type m: int
:rtype: int
"""
dp = [0] * (n + 1)
dp[1] = 0
for i in range(2, n + 1):
dp[i] = (dp[i - 1] + m) % i
return dp[n]