约瑟夫环问题
问题定义
0,1,···,n-1这n个数字排成一个圆圈,从数字0开始,每次从这个圆圈里删除第m个数字(删除后从下一个数字开始计数)。求出这个圆圈里剩下的最后一个数字。
例如,0、1、2、3、4这5个数字组成一个圆圈,从数字0开始每次删除第3个数字,则删除的前4个数字依次是2、0、4、1,因此最后剩下的数字是3。
解法
1. 模拟链表
该方法时间复杂度为O(n * m)
/**
* @author 49367
* @date 2021/4/5 10:47
*/
public class ListNode {
public int val;
public ListNode next = null;
public ListNode(int val){
this.val = val;
}
public ListNode() {}
public ListNode(int val, ListNode next) { this.val = val; this.next = next; }
}
public class Solution {
/**
* 约瑟夫环
* @param n int整型
* @param m int整型
* @return int整型
*/
public int ysf (int n, int m) {
// write code here
ListNode head = new ListNode(1);
ListNode tail = head;
for (int i = 2; i <= n; i++){
tail.next = new ListNode(i);
tail = tail.next;
}
tail.next = head;
int i = 0;
ListNode pre = tail, cur = head;
while (i < n - 1){
for (int j = 0; j < m - 1; j++){
pre = cur;
cur = cur.next;
}
pre.next = cur.next;
cur = cur.next;
i++;
}
return cur.val;
}
}
2. 数学方法(时间复杂度较低)
该方法时间复杂度为O(n)
一个(n, m)的约瑟夫环问题,假设索引从0开始,第一次删除的索引应该为 (m - 1) % n,删除后的环从 t = m % n 开始。则(n - 1, m)的约瑟夫环问题与(n, m)的约瑟夫环问题的索引的对应关系有:
则设f(n - 1)为(n - 1, m)约瑟夫环问题对应的索引,(n, m)约瑟夫环问题对应的索引就是f(n) = (f(n - 1) + t) % n = (f(n - 1) + m % n ) % n = (f(n - 1) + m) % n。
由于我们最终是要得到最后存活的索引,并且一定有f(1) = 0,即(1, m)约瑟夫环问题最后存活的索引一定是0,因此我们可以根据f(1) = 0逆推出任意约瑟夫环问题最终存活的索引。
代码如下:
class Solution {
public int lastRemaining(int n, int m) {
// 初始x = 0, 对应(1, m)约瑟夫环问题
int x = 0;
for (int i = 2; i <= n; i++) {
x = (x + m) % i;
}
return x;
}
}