题目描述
面试题62: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
思路分析
约瑟夫环问题:
方法一、使用list模拟循环链表
使用list模拟循环链表,用cur作为指向list的下标位置。
当cur移到list末尾直接指向list头部,当删除一个数后list的长度和cur的值相等则cur指向0
leecode超时,牛客通过
/*
struct ListNode{
int val;
ListNode* next;
ListNode(x):val(x),next(null);
}
*/
class Solution {
public:
int lastRemaining(int n, int m) {
if(n < 1 || m < 1) //非法输入
return -1;
//新建双向链表并初始化
list<int> nums;
for(int i = 0; i < n; ++i)
nums.push_back(i);
//构建环形链表并进行删除操作
list<int>::iterator current = nums.begin(); //迭代器初始位置
while(nums.size() > 1){
//构建第m个迭代器
for(int i = 1; i < m; ++i){
++current;
if(current == nums.end())
current = nums.begin();
}
//构建第m个迭代器的下一个迭代器
list<int>::iterator next = ++current;
if(next == nums.end())
next = nums.begin();
--current; //whlie中每次真正做的事情
nums.erase(current);
current = next; //循环变量
}
return *current; //返回最后剩下的数
}
};
链接:https://www.nowcoder.com/questionTerminal/f78a359491e64a50bce2d89cff857eb6
来源:牛客网
方法二:找规律(动态规划)
参考【1】
使用动态规划。我们注意到,输入的序列在删除一个元素后,序列的长度会改变,如果索引
在被删除的元素位置开始计算,那么每删除一个元素,序列的长度减一而索引会完全改变。
如果能找到改变前的索引和新索引的对应关系,那么该问题就容易解决了。
我们定义一个函数f(n, m),表示每次在n个数字0,1,2,3,…,n-1中每次删除第m个数字后剩下
的数字。那么第一个被删除的数字的索引是(m-1)%n。删除该索引元素后,剩下的n-1个数字
为0,1,2,…,k-1,k+1,…,n-1。下次删除数字是从k+1位置开始,于是可以把序列看
作k+1,…,n-1,0,1,…,k-1。该序列最后剩下的序列也是f的函数。但该函数和第一个函数
不同,存在映射关系,使用f’来表示,于是有:f(n, m)=f’(n-1, m)。接下来需要找到映射关系。
k+1 --> 0
k+2 --> 1
.
.
.
n-1 --> n-k-2
0 --> n-k-1
.
.
.
k-1 --> n-2
所以可以得到:right = left-k-1,则p(x) = (x-k-1)%n,而逆映射是p’(x) = (x+k+1)%n
即0 ~ n-1序列中最后剩下的数字等于(0~n-2序列中最后剩下的数字+k)%n,很明显当n=1时,
只有一个数,那么剩下的数字就是0.问题转化为动态规划问题,关系表示为:
f(n)=(f(n-1)+m)%n; 当n=1,f(1)=0;
数学归纳法证明:
递推公式:f(n,m)=[f(n-1,m)+m]%n,其中f(i,j)表示当问题规模为i,报数长度为j时最后胜利者的编号。
证明:使用数学归纳法进行证明。
当n=1时,f(1,m)=0,因为编号从0开始且只有一个人,胜利者编号显然为0。
当n=2时,序列为0,1,若m为奇数,则胜利者编号为1;若m为偶数,则胜利者编号为0,易有f(2,m)=m%2=(0+m)%2=[f(1,m)+m]%2,结论成立。
假设当n=i-1时结论成立,即对于序列0,1,2,…,i-2而言,最后的胜利者编号为f(i-1,m)。
当n=i时,序列为0,1,2,…,i-1。设第一轮的淘汰者编号为k(若m%i=0,则k=i-1,否则k=m%i-1),则序列可表示为0,1,2,…,k-1,k,k+1,…,i-1。第一轮淘汰k,余下的序列x’为k+1,…,i-1,0,1,…,k-1,问题规模变为i-1。
因为由归纳假设,当n=i-1时,对于序列x:0,1,2,…,f(i-1,m),…,i-2,胜利者编号为f(i-1,m)。由于x’=(x+k+1)%i,故f(i,m)=[f(i-1,m)+k+1]%i。当m%i=0时,k+1=i,[f(i-1,m)+k+1]%i=[f(i-1,m)+i]%i=f(i-1,m)%i+0=f(i-1,m)%i+m%i=[f(i-1,m)+m]%i;当m%i!=0时,k+1=m%i,[f(i-1,m)+k+1]%i=[f(i-1,m)+m%i]%i=[f(i-1,m)+m]%i。故当n=i时,结论成立。
综上,命题成立。
感想:因为当n=i-1时,对于序列x:0,1,2,…,f(i-1,m),…,i-2,胜利者编号为f(i-1,m)。对于问题规模为i时,经过第一轮淘汰,得到序列x’。只要能够发现x与x’间的对应关系,由f(i-1,m)就能得到f(i,m)。
作者:wandore
链接:https://leetcode-cn.com/problems/yuan-quan-zhong-zui-hou-sheng-xia-de-shu-zi-lcof/solution/guan-yu-yue-se-fu-huan-wen-ti-di-tui-gong-shi-de-z/
来源:力扣(LeetCode)
class Solution {
public:
int lastRemaining(int n, int m) {
int res=0;
for(int i=2;i<=n;i++)
res=(res+m)%i;
return res;
}
};