这题很有意思,也很巧妙,故记录下来。
官方题解思路,是约瑟夫环的数学解法:
我们将上述问题建模为函数 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 - 1, m) = (m % n + x) % n = (m + x) % n
。
我们递归计算f(n, m)
,f(n - 1, m)
,f(n - 2, m)
, ... 直到递归的终点f(1, m)
。当序列长度为1
时,一定会留下唯一的那个元素,它的编号为0
。
作者:LeetCode-Solution
链接:https://leetcode-cn.com/problems/yuan-quan-zhong-zui-hou-sheng-xia-de-shu-zi-lcof/solution/yuan-quan-zhong-zui-hou-sheng-xia-de-shu-zi-by-lee/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
需要注意的是下标的细节(老生常谈了):这里的数组 index = number,但 m 是从1开始计数。f() 的返回值 x 应该是从0开始计数的number。上文中“应当是从m % n
开始数的第x
个元素。”这句话里的m % n
是被删除的元素的下一个元素的index,从它开始的index为x
的元素正是下一步要删除的元素在原数组里的index。
Java递归代码如下:
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;
}
}
// 13 ms / 41.8 MB 37.71% / -%
- 时间复杂度:
O(n)
,需要求解的函数值有n
个。 - 空间复杂度:
O(n)
,函数的递归深度为n
,需要使用O(n)
的栈空间。
写成迭代,代码如下:
class Solution {
public int lastRemaining(int n, int m) {
int index = 0;
for (int i = 2; i <= n; i++) {
index = (index + m) % i;
}
return index;
}
}
// 7 ms / 36.6 MB 99.88% / -%
- 时间复杂度:
O(n)
,需要求解的函数值有n
个。 - 空间复杂度:
O(1)
,只使用常数个变量。
总的来说迭代的时间和空间都更少。而且迭代的写法其实有点像动态规划了。
以下是踩坑过程:
我首先想的是,建一个首尾相连的链表,然后暴力跑,可想而知的超时,只通过了26/36个测试用例。
然后想了下优化策略,每次访问所需的步数可以通过模运算减少,然后调整了一下每次访问的过程,虽然多通过了一个测试用例,但还是超时。
class Solution {
public int lastRemaining(int n, int m) {
Node head = new Node(-1);
Node p = head;
Node q = head;
for (int i = 0; i < n; i++) {
p.next = new Node(i);
p = p.next;
}
int step = 0;
while (n > 1) {
step = (step + m - 1) % n + 1;
q = head;
p = head;
for (int i = 0; i < step; i++) {
q = p;
p = p.next;
}
//System.out.println(p.value);
q.next = p.next;
step--;
n--;
}
return head.next.value;
}
}
class Node {
int value;
Node next;
Node(int value) {
this.value = value;
}
}
然后把数据存到 LinkedList 中去试试,发现效果差不多,还是超时。结果没想到换成 ArrayList 之后居然能通过,看来在数据访问和数据移动的性能取舍之间有一个微妙的平衡,本题是倾向于需要更快的访问速度。
class Solution {
public int lastRemaining(int n, int m) {
// List<Integer> list = new LinkedList<>();
List<Integer> list = new ArrayList<>();
for (int i = 0; i < n; i++) {
list.add(i);
}
int index = 0;
while (list.size() != 1) {
index = (index + m - 1) % list.size();
// System.out.println(list.get(index));
list.remove(index);
}
// System.out.println(Arrays.toString(list.toArray()));
return list.get(0);
}
}
// 1122 ms / 43.1 MB 16.61% / -%