62题-圆圈中最后剩下的数字

1 题目描述

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

2 解题思路

方法:链表模拟(超时)

  • 将[0,n]依次存储在链表中;
  • 只要链表的长度不为1,就一直循环,如果到了第m个就remove;否则将其添加到链表尾部;
  • 时间复杂度为O(mn)。
class Solution {
    
    
    public int lastRemaining(int n, int m) {
    
    
        List<Integer> list = new ArrayList<>();
        for (int i = 0;i < n;i++) {
    
    
            list.add(i);
        }
        while (list.size() > 1) {
    
    
            for (int j = 0;j < m;j++) {
    
    
                if (j != m - 1) list.add(list.get(0));
                list.remove(0);
            }
        }
        return list.get(0);
    }
}

方法:数学+递归

题目中的要求可以表述为:给定一个长度为n的序列,每次向后数m个元素并删除,那么最终留下的是第几个元素?

这个问题很难快速给出答案。但是同时也要看到,这个问题似乎有拆分为较小子问题的潜质:如果我们知道对于一个长度n-1的序列,留下的是第几个元素,那么我们就可以由此计算出长度为n的序列的答案。

算法:
我们将上述问题建模为函数f(n,m),该函数的返回值为最终留下的元素的序号。

首先,长度为n的序列会先删除第m%n个元素,然后剩下一个长度为n-1的序列。那么,我们可以递归地求解f(n-1,m),就可以知道对于剩下的n-1个元素,最终会留下第几个元素,我们设答案为x=f(n-1,m)

由于我们删除了第m%n个元素,将序列的长度变为n-1。当我们知道了f(n-1,m)对应的答案x之后,我们也就可以知道,长度为n的序列最后一个删除的元素,应当是从m%n开始数的第x个元素。因此有f(n,m)=(m%n+x)%n=(m+x)%n

class Solution {
    
    
    public int lastRemaining(int n, int m) {
    
    
        return f(n,m);
    }
    private int f(int n, int m) {
    
    
        if (n == 1) return 0;
        int x = f(n-1,m);
        return (x + m) % n;
    }
}

复杂度分析:

扫描二维码关注公众号,回复: 12566150 查看本文章
  • 时间复杂度:O(n),需要求解的函数值有 n 个。
  • 空间复杂度:O(n),函数的递归深度为 n,需要使用 O(n) 的栈空间。

猜你喜欢

转载自blog.csdn.net/qq_41242680/article/details/113534229