set中要注意的问题

#include<stdio.h>
#include<set>
#include<iostream>
using namespace std;
set<int>a;
set<int>::iterator it,it1;
int main()
{
	int n;
	cin >> n;
	while (n--)
	{
		int m;
		cin >> m;
		if(m==1)
            {cout<<1<<endl;
            continue;
            }
        else if(m==2)
        {
            cout<<"1 2"<<endl;
            continue;
        }
        else if(m==3)
        {
            cout<<"1 2 3"<<endl;
            continue;
        }
		int i;
		for (i = 1; i <= m; i++)
		{

			if (i % 2 != 0)
			{
				a.insert(i);
			}
		}
		int j, k = 0;
		for (i=0; a.size()>3;i++)
		{
			k = 0;
			if (i % 2 == 0)
			{

					for (it = a.begin(); it != a.end(); it++)
					{
						k++;

						if (k % 3 == 0)
						{
							it1 = ++it;
							it--;
							a.erase(it);
							it = it1;
							it--;
						}
					}

			}
			else
			{

					for (it = a.begin(); it != a.end(); it++)
					{
						k++;
						if (k % 2 == 0)
							{

						    it1 = ++it;
							it--;
							a.erase(it);
							it = it1;
							it--;

							}
					}

			}
		}
		i = 0;
		for (it = a.begin(); it != a.end(); it++)
		{

			i++;
			if (i == 1)
				cout << *it;
			else
				cout << " " << *it;
		}
		cout << endl;
	}
}

set中若erase了迭代器,迭代器就消失了,因此我用了一个临时变量来记录一下it后面元素的位置,以便找回it,而vector则可以接着指向下一个元素,

set中不支持begin()+i(i为一个实数),但是vector中支持,一般无序容器(如map,list)都不支持,且set不能用【】来取值。

上面代码的题目

某部队进行新兵队列训练,将新兵从一开始按顺序依次编号,并排成一行横队,训练的规则如下:从头开始一至二报数,凡报到二的出列,剩下的向小序号方向靠拢,再从头开始进行一至三报数,凡报到三的出列,剩下的向小序号方向靠拢,继续从头开始进行一至二报数。。。,以后从头开始轮流进行一至二报数、一至三报数直到剩下的人数不超过三人为止。

猜你喜欢

转载自blog.csdn.net/weixin_42165786/article/details/81055148