题目描述
每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0…m-1报数…这样下去…直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!_)。请你试着想下,哪个小朋友会得到这份礼品呢?(注:小朋友的编号是从0到n-1)
如果没有小朋友,请返回-1
题解
题目抽象:给定一个由[0…n-1]构成的数组,第一次从0开始数m个数,然后删除,以后每次都从删除的数下一个位置开始数m个数,然后删除,直到剩余一个数字,找出那个数字。
比如:arr = [0 1 2 3 4], m = 3
第一次:删除2 ,变成 arr = [0 1 3 4]
第二次,删除0,变成 arr = [1 3 4]
第三次,删除4,变成 arr = [1 3]
第四次,删除1,变成 arr = [3]
方法一:模拟
- 建立链表,进行模拟
- 时间复杂度:O(N2)
- 空间复杂度:O(N)
方法二:递归
- 假设(n, m)表示最终留下元素的序号,比如上面例子中表示为:f(5,3) = 3
- 长度为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
- 而当n=1是,知道f(n,m)=0
- 时间复杂度:O(N)
- 空间复杂度:O(N)
public int LastRemaining_Solution(int n, int m) {
if (n<=0) return -1;
return dfs(n,m);
}
public int dfs(int n, int m){
if (n == 1) return 0;
int x = dfs(n-1, m);
return (x+m)%n;
}
方法三:迭代法
- 根据方法二可知:
- f[1] = 0
- f[2] = (f[1] + m) % 2
- f[3] = (f[2] + m) % 3
- …
- f[n] = (f[n]+m)%n
public int LastRemaining_Solution1(int n, int m) {
if (n<=0)return -1;
int ans = 0;
for (int i = 2; i <= n; i++){
ans = (ans + m) % i;
}
return ans;
}
- 时间复杂度:O(N)
- 空间复杂度:O(1)