约瑟夫环问题的两种解法:
目录
1.模拟
说到模拟这就简单了对吧,首先来个list的迭代器版本的:
class Solution {
public:
int lastRemaining(int n, int m) {
list<int> numbers;
for(int 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;
numbers.erase(current);
current=next;
}
return *current;
}
};
好家伙结果超时了。。。原因应该是我憨憨直接遍历找第m个吧。。。
然后来个多过了两个栗子的vector:
class Solution {
public:
int lastRemaining(int n, int m) {
vector<int> v;
for(int i = 0; i < n; i++){
v.emplace_back(i);
}
int index = 0;
while(v.size() > 1){
index = (index + m) % v.size() - 1;
if(index == -1) index = v.size() - 1;
v.erase(v.begin()+index);
}
return v[0];
}
};
还是有8个没过。
2.数学
详细可以看看原书第二版的P302,有详细的证明,涉及到一点点数论和反函数的东西,也不是很复杂,但是往这方面想确实很难。。。
证明的一点摘要如下:
首先设f(n,m)是在n个数字里每次删除第m个数最后剩下的数字。
设k=(m-1)%n,显然,k是第一次的删除的第m个数字,之后剩下0,1,...k-1,k+1...,n-1,n,然后我们将这个序列重排成k+1,...n-1,n,0,1...k-1
那这个序列叫什么呢?叫f(n-1,m)吗?不对,因为f(n-1,m)的样子应该是0,1,...n-1的一个序列,所以我们给这个重排的序列另外取一个名字叫做f'(n-1,m)。还记得f的含义么?没错,是按照规律删除的剩下的最后一个数,f'的定义类似。很容易我们想到f(n,m)=f'(n-1,m)
不过我们是要编程啊,最好可以找到一定的规律,诸如递归的形式。
原书设计了一个很巧妙的映射,将上述重排的序列映射到0,1,2...n-1
k+1->0 k+2->1.... 0->n-k-1 ... k-1->n-2
这个映射函数其实很容易求得,就是p(x)=(x-k-1)%n,逆映射是(x)=(x+k+1)%n
现在接着看映射后的序列,我们之前讲过,0,1,...n-1就是f(n-1,m)对吧,他与f'(n-1,m)的关系是不是直接代入函数就行了?
这样我们得出:
关于n=1为什么是0,应为序列开始只有一个0,所以剩下的数字显然就是0。
代码如下所示:
class Solution {
public:
int lastRemaining(int n, int m) {
int last=0;
for(int i=2;i<=n;++i)
{
last=(last+m)%i;
}
return last;
}
};
数学yyds!