算法学习
今天写pta上的练习题,用c++,stl中的vector容器实现了约瑟夫退圈问题。在使用容器时注意erase的范围,需要判断两次迭代器是否越界。否则erase函数会崩。
代码如下
#include<iostream>
#include<vector>
using namespace std;
int main()
{
int n,m;
cin>>n>>m;
while(n+m>0)
{
vector<int>ves;
for(int i=1;i<=n;i++)
{
ves.push_back(i);
}
int i=0;
while(ves.size() > 1)
{
int j=1;
while(j!=m)
{
if(i == ves.size())
{
i=0;
}
i++;
j++;
if(i == ves.size())
{
i=0;
}
}
ves.erase(ves.begin()+i);
}
cout<<ves[0]<<endl;
ves.pop_back();
cin>>n>>m;
}
return 0;
}