题目:0,1,...n-1这n个数字排成一个圆圈,从数字0开始每次从这个圆圈里删除第m个数字。求出这个圆圈里剩下的最后一个数字。
分析:例如,0、1、2、3、4这5个数字组成一个圆圈,从数字0开始每次删除第3个数字,则删除的前四个数字依次是2、0、4、1,因此最后剩下的数字是3。这个是有名的约瑟夫环问题。我们介绍两种方法:一种方法是用环形链表模拟圆圈的经典解法,第二种方法是分析每次被删除的数字的规律并直接计算出圆圈中最后剩下的数字。
解法一:经典解法,用环形链表模拟圆圈
创建一个总共有n个结点的环形链表,然后每次在这个链表中删除第m个结点。如果有要求不能使用标准模板库里面的数据容器来模拟环形链表,我们就自己实现。如果面试官没有特殊要求,我们就可以用模板库里的std::list来模拟一个环形链表。由于std::list本身并不是一个环形结构,因此每当迭代器扫描的链表末尾的时候,要记得把迭代器移动到链表的头部,这样就相当于按照顺序在一个圆圈中遍历了。代码如下:
int LastRemaining(unsigned int n,unsigned int m)
{
if(n<1||m<1)
return -1;
unsigned int i=0;
list<int> numbers;
for(i=0;i<n;i++)
numbers.push_back(i);
list<int>::Iterator current=numbers.begin();
while(numbers.size()>1)
{
for(int i=1;i<m;i++)
{
current++;
if(current==numbers.end())
current=numbers.begin();
}
list<int>::Iterator next=++current;
if(next==numbers.end())
next=numbers.begin();
--current;
list.erase(current);
current=next;
}
return *(current);
}
解法二:创新的解法
思路:
假设从0-n-1中删除了第m个数字,则下一轮的数字排列为m,m+1,.....n,1,2,3...m-2,将该数字排列重新映射为0~n-2,则为m 0
m+1 1
.... ....
n-1 n-1-m
0 n-m
1 n-m+1
... ....
m-2 n-2
可以看出从右往左的映射关系为left=(right+m)%n,即0~n-1序列中最后剩下的数字等于(0~n-2序列中最后剩下的数字+m)%n,很明显当n=1时,只有一个数,那么剩下的数字就是0.
问题转化为动态规划问题,关系表示为:
f(n)=(f(n-1)+m)%n; 当n=1,f(1)=0;
代码如下:
int LastRemaining(unsigned int n,unsigned int m)
{
if(n<1||m<1)
return -1;
int last=0;
for(int i=2;i<=n;i++)
last=(last+m)%i;
return last;
}
测试用例:
功能测试(输入的m小于n,输入的m大于或者大于n)
特殊输入测试(圆圈中有0个数字)
性能测试(从最初有4000个数字的圆圈中每次删除第997个数字)