第46题 孩子们的游戏(圆圈中最后剩下的数)
题目描述
每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0…m-1报数…这样下去…直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!_)。请你试着想下,哪个小朋友会得到这份礼品呢?(注:小朋友的编号是从0到n-1)
如果没有小朋友,请返回-1
思路:
前面删除了m个数,index + m 在前面补m个数,然后模 i 得到上一轮的 index
链表模拟
class Solution {
public:
int LastRemaining_Solution(int n, int m)
{
if(n<=0) return -1;
list<int> ls;
for(int i =0;i<n;++i){
ls.push_back(i);
}
int index = 0;
while(n>1){
index = (index+m-1) % n;
auto bg = ls.begin();
std::advance(bg,index);
ls.erase(bg);
--n;
}
return ls.back();
}
};
递归
class Solution {
public:
int f(int n,int m){
if(n==1) return 0;
return (f(n-1,m)+m)%n;
}
int LastRemaining_Solution(int n, int m)
{
if(n<=0) return -1;
return f(n,m);
}
};
n个数时,删除一个数(位置在m-1)后,状态变成n-1个数从m处开始的
f
(
n
−
1
,
m
)
f(n-1,m)
f(n−1,m)状态,所以如果
f
(
n
,
m
)
f(n,m)
f(n,m)表示n个数从0位置开始的最后剩的数结果,那么有转移方程:
f
(
n
,
m
)
=
(
f
(
n
−
1
,
m
)
+
m
)
m
o
d
n
f(n,m) = (f(n-1,m)+m) \mod n
f(n,m)=(f(n−1,m)+m)modn
其中加m表示起点偏移m,mod n 表示当前数组长度为n内index
迭代
class Solution {
public:
int LastRemaining_Solution(int n, int m)
{
if(n<=0) return -1;
int index = 0;
for(int i = 2;i<=n;++i){
index = (index+m)% i;
}
return index;
}
};
class Solution:
def lastRemaining(self, n: int, m: int) -> int:
index = 0
for i in range(2, n + 1):
index = (index + m) % i
return index