这篇文章主要介绍了LeetCode如何求圆圈中最后剩下的数字,具有一定借鉴价值,感兴趣的朋友可以参考下,希望大家阅读完这篇文章之后大有收获,下面让小编带着大家一起了解一下。
题目描述
0,1,,n-1 这 n 个数字排成一个圆圈,从数字 0 开始,每次从这个圆圈里删除第 m 个数字。求出这个圆圈里剩下的最后一个数字。
例如,0、1、2、3、4 这 5 个数字组成一个圆圈,从数字 0 开始每次删除第 3 个数字,则删除的前 4 个数字依次是 2、0、4、1,因此最后剩下的数字是 3。
1 <= n <= 10^5 1 <= m <= 10^6
题目样例
示例
题目思考
n 个数字和 n-1 个数字最后剩下的数字有什么关系吗?
解决方案
思路
一个比较容易想到的思路是暴力法, 即使用数组模拟整个过程: 记录当前起点, 求要移除的下个数字的下标, 并从原列表中移除它. 但这样时间复杂度达到了 O(N^2), 因为移除元素也需要 O(N) 时间 而改用双端链表的话则为 O(MN), 因为虽然移除变为了 O(1), 但每次定位下一个要移除的数字下标时不能像数组那样 O(1) 时间定位, 而是需要 O(M) 时间来遍历到下个移除的位置 那有什么方法可以做到更小的时间复杂度呢? 如果我们能从 n-1 个数字的结果, 推导出 n 个数字的结果, 那就可以一次遍历 O(N) 时间搞定 所以尝试动态规划的思路, 设 dp[n, m]
是 n 个数字时每次删除第 m 个数字的最后剩下数字的下标第一次删除时, 删除数字的下标 i 为 (m-1)%n
, 而其下一个下标 i+1 为下一次删除的起点此时将新的起点 i+1 映射为下标 0, 自然 i+2 就对应下标 1, ... 到最后原下标 i 就对应映射下标 n-2. 这样就转换成了 n-1 个数字时每次删除第 m 个数字的情况 显然映射下标 x 与原下标 y 的关系是 y = (x+i+1)%n
, 而 i 为(m-1)%n
, 所以可以进一步得到y = (x+m)%n
下标映射之后, 此时总共有 n-1 个数字, 最终剩余的数字就是 dp[n-1,m]
, 这个是映射下标, 根据上面的下标转换关系, 反推出原下标就是(dp[n-1,m]+m)%n
而不管是第一次删除还是第二次删除, 其最终剩余的数字只能是同一个, 所以 dp[n, m] = (dp[n-1,m]+m)%n
, 这就是最核心的状态转移方程观察可以发现, 每个 dp 值只和前一个 dp 值相关, 所以我们可以使用一个变量来节省空间, 然后从小到大遍历到 n 即可 而 dp 初始值为 n=1 的情况, 此时最后剩余的数字显然只能是 0
复杂度
时间复杂度 O(N): 只需要遍历到 N 空间复杂度 O(1): 只使用了常数个变量
代码
class Solution:
def lastRemaining(self, n: int, m: int) -> int:
# 初始dp[1,m]为0
dp = 0
for x in range(2, n + 1):
# 对应公式dp[x,m] = (dp[x-1,m] + m) % x
dp = (dp + m) % x
return dp
感谢你能够认真阅读完这篇文章,希望小编分享的“LeetCode如何求圆圈中最后剩下的数字”这篇文章对大家有帮助,同时也希望大家多多支持编程之家,关注编程之家行业资讯频道,更多相关知识等着你来学习!
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 [email protected] 举报,一经查实,本站将立刻删除。