原题来自牛客网剑指offer:链接
题目描述
每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0…m-1报数…这样下去…直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!_)。请你试着想下,哪个小朋友会得到这份礼品呢?(注:小朋友的编号是从0到n-1)
如果没有小朋友,请返回-1
思路
约瑟夫环问题,直接模拟过程即可。
法1,使用数组记录某个小朋友是否仍在环中,用数组模拟环。
class Solution {
public:
int LastRemaining_Solution(int n, int m)
{
if(n==0)
return -1;
int nownum=0,nowpos=0,nowrest=n,out[10000]={0};
while(nowrest!=1){
if(!out[nowpos]){
if(nownum==m-1){
out[nowpos]=1;
nownum=-1;
nowrest--;
}
nownum++;
}
if(nowrest!=1)
nowpos=(nowpos+1)%n;
}
for(int i=0;i<n;i++){
if(!out[i])
return i;
}
return -1;
}
};
法2,使用STL的链表list,小朋友出圈了就直接erase掉,迭代器到了end就回到begin。
class Solution {
public:
int LastRemaining_Solution(int n, int m)
{
if(n==0)
return -1;
list<int> children;
int now=0;
for(int i=0;i<n;i++)
children.push_back(i);
auto iter=children.begin();
while(children.size()>1){
if(now==m-1)
iter=children.erase(iter);
else
iter++;
now=(now+1)%m;
if(iter==children.end())
iter=children.begin();
}
return *(children.begin());
}
};