题目
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个元素。
public int lastRemaining(int n, int m) {
List<Integer> list = new ArrayList<Integer>(n);
for (int i = 0; i < n; i++) {
list.add(i);
}
int index = 0;
while (n > 1) {
index = (index + m -1)%n;
list.remove(index);
n--;
}
return list.get(0);
}
时间复杂度:O(n^2) remove的复杂度为O(n) 所有整体复杂度为平方
空间复杂度:O(n) 链表的内存空间
动态规划
动态规划三步曲:
第一步:找状态。题意为在数字n中每隔第m个数字删除一个数字,并从下一个数字开始又重复,直到剩余最后一个数字结束;首先分析得出第一次删除的数字为:
(m-1)%n // 防止m数字比n大,所以取余
下一次开始的数字是:
m%n // 防止m数字比n大,所以取余
我们假设下一次开始的数字为t,即
t = m % n
那么,下一次开始的整个数字集合为:
t, t+1, t+2, t+3,... ,0, 1, 2, 3, ..., t-2
我们用一维数组dp的下标表示n,dp[n]的值则表示最后一个剩余的值。
第二步:找状态转移方程式。第一步分析了数字n删除一轮后的数字集合,我们需要找到数字(n-1)最后剩下的值;可以对比下面的数字集合:
数字n-1的集合: 0, 1, 2, 3, ...,n-3, n-2
数字n删除第一轮后的集合: t+0, t+1, t+2, t+3, ..., t-2
可以看出,数字n-1的集合与数字n删除第一轮后的集合存在一定的关系,也就可以根据数字n-1最后剩余的数字求出数字n最后剩余的数字,我们假设数字(n-1)最后剩余的数字是X,那么数字n最后剩余的数字是:
(x+t)%n // 防止m数字比n大,所以取余
所以可以得出状态转移方程式:
dp[n] = (dp[n-1]+m%n)%n
第三步:找出边界条件。很显然,dp[1] = 0恒成立。
分析完毕,下面给出实现代码:
public int lastRemaining(int n, int m) {
int x = 0;
for (int i = 2; i <= n; i++) {
x = (x + m)%i;
}
return x;
}
时间复杂度:O(n) 数字n的大小
空间复杂度:O(1)
注意:这个问题也是约瑟夫环问题。