【LeetCode-剑指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

二、解决

1、暴力模拟（勉强通过）

思路：

当前删除位置idx，下一个删除位置是idx + m。
但是，由于当前位置数字删除，后面数字会前移一位，所以实际下一个位置是idx + m - 1。数字到末尾会从头开始数，最后取模下，就是(idx + m - 1) % n。

附：ArrayList.remove()代码

• LinkedList.remove() VS ArrayList.remove()
  LinkedList的remove()时间复杂度是$O(n)$，与ArrayList相同。但ArrayList是内存连续空间拷贝，相比LinkedList大量非连续地址访问，ArrayList性能很OK。

代码：

class Solution {
    public int lastRemaining(int n, int m) {
        ArrayList<Integer> list = new ArrayList<>(n);
        for (int i = 0; i < n; i++) {
            list.add(i);
        }
        int idx = 0;
        while (n > 1) {
            idx = (idx + m - 1) % n;
            list.remove(idx);
            n--;
        }
        return list.get(0);
    }
}

时间复杂度： $O(n^2)$
空间复杂度： $O(n)$

2、数学解法

思路：

1、这题是著名约瑟夫环问题，有数学解法。

全文重点提示：只关心最终活着的那个人的序号变化。

2、问题：给每个人一个编号（索引值），每个人用字母代替。
下面是N=8，m=3的例子：F(n, m)表示为最后剩下那个人的索引号，因此只关心最后剩下来这个人的索引号的变化情况即可：

从8个人开始，每次杀掉一个人（位置为index），去除被杀的人（index），然后把那个人之后的第一个人（index+1）作为开头，重新编号。

• 第一次C被杀掉，人数变成7，D作为开头（最终活下来的G的编号从6变成3）
• 第二次F被杀掉，人数变成6，G作为开头（最终活下来的G的编号从3变成0）
• 第三次A被杀掉，人数变成5，B作为开头（最终活下来的G的编号从0变成3）
• 以此类推，当只剩一个人时，他的编号必定为0！（重点）

3、最终活着的人编号反推

知道了G的索引号的变化过程，那反推一下，从 N=7 到 N=8 的过程。

由上可得：$f(8,3)=[f(7,3)+3]\%8$
泛化推广，即$f(n,m)=[f(n-1,m)+m]\%m$

4、递推公式导出

$$f(n,m)=\{\begin{array}{rlr}0& & \text{ n=0}\\ [f(n-1,m)+m]\%n& & \text{ n>1}\end{array}$$

更有逻辑的推导可参看7-28 猴子选大王 (20 分)，非常严格的数学推导可以看关于约瑟夫环问题递推公式的证明。

代码：

class Solution {
    public int lastRemaining(int n, int m) {
        int ans = 0;
        // 最后一轮剩下2个人，所以从2开始反推
        for (int i = 2; i <= n; i++) {
            ans = (ans + m) % i;
        }
        return ans;
    }
}

时间复杂度： $O(n)$
空间复杂度： $O(1)$

三、参考

1、Java解决约瑟夫环问题，告诉你为什么模拟会超时！
2、换个角度举例解决约瑟夫环