问题
思路
约瑟夫环的问题。
具体做法,考虑到删除,最好的办法是用循环链表。
但是,也没关系,用STL list即可。
小心迭代器失效的问题!!!其实,链表只是当前元素可能失效,但是迭代器可能到无效位置。
代码
class Solution {
public:
int LastRemaining_Solution(int n, int m)
{
if( n < 1 || m < 1) return -1;
list<int> nums;
for(int i = 0; i < n; ++i){
nums.push_back(i);
}
typedef list<int>::iterator iter;
iter it = nums.begin();
while(nums.size() > 1){
// move forward m-1 step
for(int k = 0; k < m-1; ++k){
++it;
if( it == nums.end() )
it = nums.begin();
}
// delete the mth one
it = nums.erase(it);
if( it == nums.end() ) it = nums.begin();
}
return *it;
}
};