士兵队列训练问题

Hdu-1276

题目描述

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

输入

本题有多个测试数据组，第一行为组数N，接着为N行新兵人数，新兵人数不超过5000。

输出

共有N行，分别对应输入的新兵人数，每行输出剩下的新兵最初的编号，编号之间有一个空格。

示例输入

2

20

40

示例输出

1 7 19

1 19 37

代码

#include<iostream>
#include<vector>
using namespace std;
int main()
{
	int n, t;
	cin >> t;
	while (t--)
	{
		cin >> n;
		int k = 2;
		vector<int> mylist;
		vector<int>::iterator it;
		for (int i = 1; i <= n; i++)
			mylist.push_back(i);
		while (mylist.size() > 3)
		{
			int num = 1;
			for (it = mylist.begin(); it != mylist.end();)
			{
				if (num++ % k == 0)
					it = mylist.erase(it);
				else
					it++;
			}
			k == 2 ? k = 3 : k = 2;
		}
		for (it = mylist.begin(); it != mylist.end(); it++)
		{
			if (it != mylist.begin())
				cout << " ";
			cout << *it;
		}
		cout << endl;
	}
	return 0;
}