题目描述:
每年六一儿童节,牛客都会准备一些小礼物去看望孤儿院的小朋友,今年亦是如此。HF作为牛客的资深元老,自然也准备了一些小游戏。其中,有个游戏是这样的:首先,让小朋友们围成一个大圈。然后,他随机指定一个数m,让编号为0的小朋友开始报数。每次喊到m-1的那个小朋友要出列唱首歌,然后可以在礼品箱中任意的挑选礼物,并且不再回到圈中,从他的下一个小朋友开始,继续0…m-1报数….这样下去….直到剩下最后一个小朋友,可以不用表演,并且拿到牛客名贵的“名侦探柯南”典藏版(名额有限哦!!^_^)。请你试着想下,哪个小朋友会得到这份礼品呢?(注:小朋友的编号是从0到n-1)
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int LastRemaining_Solution(int n, int m)
{
if(n < 1 || m <1)
return -1;
if(n == 1)
return 0;
vector<int> children;
for(int i = 0; i < n; i++){
children.push_back(i);
}
vector<int>::iterator current = children.begin();
while(children.size() != 1){
for(int i = 1; i < m; i++){
current++;
if(current == children.end()){
current = children.begin();
}
}
children.erase(current); //注意erase的用法,erase current指向的元素之后,current会自动指向当前元素的下一元素;
if(current == children.end()) //鉴于上一语句erase了一个元素,而下一元素有可能是end(),所以要增加这个判断,否则会段错误;
current = children.begin();
}
return children[0];
}
};
int main()
{
Solution s;
int n = s.LastRemaining_Solution(5, 1);
cout << n << endl; //n = 4;
return 0;
}
解体思想很简单,就是模拟真实的场景,需要注意的两点:
1、取迭代器指向的值,可以像指针一样使用 *
2、erase 迭代器it指向的一个元素之后,it会自动的指向下一元素;
原因是这样的:引自这里
因为vector容器是一个连续的内存块,vector::iterator其实就是当前元素所指向的地址,它不同于list或map的iterator,所以删除还有一种更直接的方法,如下:
vector v;
假如我们需要删除v中第5个元素
v.erase( (vector::iterator)&v[5] );