LeetCode如何求圆圈中最后剩下的数字

这篇文章主要介绍了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 = 5, m = 3

  • 输出: 3

  • 输入: n = 10, m = 17

  • 输出: 2

                     

题目思考

  1. 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如何求圆圈中最后剩下的数字”这篇文章对大家有帮助,同时也希望大家多多支持亿速云,关注亿速云行业资讯频道,更多相关知识等着你来学习!

原创文章,作者:1402239773,如若转载,请注明出处:https://blog.ytso.com/tech/opensource/223574.html

(0)
上一篇 2022年1月7日 00:43
下一篇 2022年1月7日 00:43

相关推荐

发表回复

登录后才能评论