题目描述:
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、ArrayList遍历 + 模拟链表:
可以使用一个ArrayList来模拟链表,当删除位置index
处的数字,下一个要删除的数字位置为(index + m - 1) % n
(由于删除一个数字后,后面的数字位置要往前移动一位,因此是m - 1
,同时为了避免超出索引,因此要进行取余操作)。
由于删除数字过程中链表中的数字个数会不断减少,因此每删除一个数字,都要更新n: n--
。循环上述操作,链表中剩下唯一一个数字,该数字就是最后一个数字。
注1: 使用LinkedList
来模拟会超时,这是因为LinkedList
虽然执行remove
操作的时间复杂度为O(1)
,但是查找删除的位置所需的时间复杂度为O(n)
,因此LinkedList单次删除的时间复杂度和ArrayList是一样的。同时,由于ArrayList 的 remove 操作在后续移位的过程中,其实是内存连续空间的拷贝的!因此相比于LinkedList大量非连续性地址访问,ArrayList的性能是很 OK 的!
注1: 以上分析参考于sweetieeyi的题解:Java解决约瑟夫环问题,告诉你为什么模拟会超时!
2、数学分析:
这个思路和解法1有点类似,设f(n,m)
为在n
个数字中删除第m
个数字最后剩下的数字的位置。
从解法1的思路可以看出,当n
个人删除一个数字后,之后的数字向前移动了m
位,那么所求的最后剩下的数字winner
也向前移动m
位,即这个过程winner的位置变化就是f(n,m) -> f(n - 1,m)
;
将这个过程逆过来看,如果已知f(n -1, m)
,即n -1
个人的winner
的位置,那么在n
个人时winner
的位置应该往后移动m
位,即f(n,m) = (f(n -1,m) + m) % n,n > 1
。
初始条件: 当n = 1
时,显然f(n,m) = 0
。
注2: 以上分析参考于约瑟夫环——公式法(递推公式)!
时间复杂度和空间复杂度: 解法1的时间复杂度为O(N^2),空间复杂度为O(N);解法2的时间复杂度为O(N),空间复杂度为O(1)。
实现代码:
//解法1:ArrayList遍历 + 模拟链表
public int lastRemaining(int n, int m) {
if(n < 1 || m < 1)
return -1;
ArrayList<Integer> res = new ArrayList<>();
for(int i = 0; i < n; i++)
res.add(i);
int index = 0;
while(n > 1){
index = (index + m - 1) % n;
res.remove(index);
n--;
}
return res.get(0);
}
//解法2:数学分析
public int lastRemaining1(int n, int m) {
if(n < 1 || m < 1)
return -1;
int res = 0;
for(int i = 2; i <= n; i++)
res = (res + m) % i;
return res;
}