今天去做了个笔试题,有一个经常遇到的题:有20个人围成一圈,从第一个人开始,每数到3,这个人就退出,后面的人接着从1开始数,一直循环下去,直到剩下最后一个人。
很明显用一个循环链表就可以实现,每次删除一个人的时间复杂度为O(1),但是当时想用C++的标准模板库的list容器实现时,脑子一抽,觉得list没有直接实现循环功能,于是傻不拉几的用C语言开始自己写起循环链表来了,真是太浪费了。这里实现了一下,就是用迭代器遍历这个链表,当迭代器指向list.end()即最后一个元素的下一个元素时,将它指向第一个元素即可,这样就可以循环遍历整个链表了。
#include <iostream>
#include <list>
using namespace std;
void FindNPeople(int n, int m)
{
if( n <= 0 || m <= 0)
return;
list<int> Tlist;
for(int i=1; i <= n; i++)
{
Tlist.push_back(i);
}
int count = m;
list<int>::iterator iter = Tlist.begin();
while(Tlist.size() != 1)
{
list<int>::iterator iterend = Tlist.end();
while(--count > 0)
{
iter++;
if(iter == iterend)
{
iter = Tlist.begin(); //构成循环链表
}
}
count = m;
iter = Tlist.erase(iter);//iter指向被删除元素的下一个元素
if(iter == iterend)
{
iter = Tlist.begin();//构成循环链表
}
}
if(Tlist.size() == 1)
{
cout<<Tlist.front()<<endl;
}
else
{
cout<<"error"<<endl;
}
}
int main()
{
FindNPeople(20, 3);
return 0;
}