题意
0,1,n-1这n个数字排成一个圆圈,从数字0开始,每次从这个圆圈里删除第m个数字。求出这个圆圈里剩下的最后一个数字。
例如,0、1、2、3、4这5个数字组成一个圆圈,从数字0开始每次删除第3个数字,则删除的前4个数字依次是2、0、4、1,因此最后剩下的数字是3。
示例
输入: n = 5, m = 3
输出: 3
输入: n = 10, m = 17
输出: 2
思路
约瑟夫环是一个经典问题,有很多种解法,本文记录的通过示例找出规律算出下标。
核心部分就在于下标:
第一次: k = m-1 % n ( 坐标 = m位置 % 总长度 )
依次坐标: k + m-1 % n ( 坐标 = 前一次坐标 + m位置 % 总长度 )
假设 0-1-2-3-4-5 ,m为3
1·顺序 0-1-[2]-3-4-5, 套用公式, k = 3-1 % 6 = 3
2·顺序 0-1-3-4-[5],套用公式, k = 3 + 3-1 % 5 = 0
3·顺序 0-1-[3]-4,套用公式, k = 0 + 3-1 % 5 = 3
解答
class Solution {
public int lastRemaining(int n, int m) {
// array 转 arrayList 为了做操作
List<Integer> nList = new ArrayList<>();
for(int i=0; i<n; i++){
nList.add(i);
}
// 算出每次排除的数字坐标
int removeIndex = (m-1) % n;
// 剩下最后一个数字 取出
while(nList.size() != 1){
nList.remove(removeIndex);
removeIndex = (removeIndex + m -1) % nList.size();
}
// 取出
return nList.get(0);
}
}