题目描述
每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0...m-1报数....这样下去....直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!^_^)。请你试着想下,哪个小朋友会得到这份礼品呢?(注:小朋友的编号是从0到n-1)
如果没有小朋友,请返回-1
思路分析
- 数组实现
利用下标遍历
int LastRemaining_Solution(int n, int m)
{
vector<int> a;
for(int i=0;i<n;++i)
a.push_back(i);
if(n<1 || m<1)
return -1;
//i遍历圈,如果到头,i赋为0
//step记录m的步数
//count小盆友的个数
int i=-1,step=0,count=n;
while(count>0){
i++;
//如果一圈走完,i从0开始
if(i>=n)
i=0;
//出圈小盆友赋值-1
if(a[i]==-1)
continue;
step++;
if(step==m){
a[i]=-1;
step=0;
count--;
}
}
return i;
}
- 递归实现
长度为n的序列会先删除第m%n个元素,剩下长度n-1的序列。递归求解f(n-1,m)
f(n-1,m)结果假设x,下一个删除的元素是从m%n开始的第x个元素。
f(n, m) = (m % n + x) % n = (m + x) % n
int LastRemaining_Solution(int n, int m)
{
if(n==0)
return -1;
if(n==1)
return 0;
return (LastRemaining_Solution(n-1,m)+m)%n;
}
- 链表实现
struct Node {
int val;
struct Node *next=NULL;
TreeNode(int x) :
val(x), next(NULL) {
}
};
int LastRemaining_Solution(int n, int m)
{
if(n<=0 || m<=0) return -1;
//将编号圈起来,构成环形链表
Node* Head=new Node(0); //编号为0的小盆友
Node* headp1=Head;
for(int i=0;i<n;i++){
headp1->next=new Node(i);
headp1=headp1->next;
}
headp1->next=head;
//定义新节点模拟小盆友数数
Node* headp2=head;
Node* pre=NULL;
int count=1;
while(True){
//走了m个数,删除结点
//结束条件,只剩两个结点
if(count==m){
if(head2->next==pre)
return pre->val;
}
//删除节点
Node* temp=headp2->next;
pre->next=temp;
head ->next=NULL;
headp2=temp;
count=1;
}
else{
pre=headp2;
headp2=headp2->next;
count++;
}
}