巧解约瑟夫环问题(C/C++)
问题描述:N个人围成一圈,从第一个开始报数,第M个将被杀掉,最后剩下一个,其余人都将被杀掉。例如N=6,M=5,被杀掉的顺序是:5,4,6,2,3,1。
分析以上被杀掉的顺序:
第一轮:[1,2,3,4,5,6] 杀死序号为5的人,下一轮从序号为6的人开始;
第二轮:[1,2,3,4,6] 杀死序号为4的人,下一轮从序号为6的人开始;
第三轮:[1,2,3,6] 杀死序号为6的人,下一轮从序号为1的人开始;
第四轮:[1,2,3] 杀死序号为2的人,下一轮从序号为3的人开始;
第五轮:[1,3] 杀死序号为3的人,剩下一人序号为1,存活。
方法一(普通解法):
构造一个 int 型向量其大小为N,依次存放[1,2,…,n],从下标为0(也就是第一个人)开始搜寻,搜寻到第M个人,将其移除,依次循环直至剩余1人(这时向量的大小为1),将此向量存放的值返回,即存活下来的人的序号。
代码如下:
int lastRemaining(int n, int m){
int cur_delete = -1; //当前将要杀死的人的下标
vector<int> vec;
for(int i=1;i<=n;i++){
vec.push_back(i);
}
while(n != 1){
cur_delete += m; //从当前位置向后寻找第M个人
cur_delete %= n; //得到第M个人在循环中的下标
vec.erase(vec.begin()+cur_delete); //将其杀死!
--cur_delete; //杀死之后坐标向前移一位
--n;
}
return vec[0];
}
该代码虽能解决问题,但是由于向量中每次执行erase()方法的时间复杂度为O(n),当输入的数很大时,会出现超时的情况:
方法二(递归):
我们将上述问题转换成一个函数 f ( n , m ) f(n,m) f(n,m),该函数的返回值是最终存活的人的下标:
首先,长度为n的序列会先删除第 m%n 个元素,然后剩下长度为n-1的序列,我们可以递归地求解f(n-1,m),就可以知道剩下的n-1个元素中最后留下第几个元素,设答案是 f ( n − 1 , m ) f(n-1,m) f(n−1,m)。
f ( n , m ) = { 0 , n==1 ( ( m % n ) + f ( n − 1 , m ) ) % n , n>1 f(n,m)= \begin{cases} 0, &\text {n==1} \\ ((m\%n)+f(n-1,m))\%n,&\text{n>1} \end{cases} f(n,