士兵队列训练问题(STL — 链表练习)

原题链接

Problem Description

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

Input

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

Output

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

Sample Input

2
20
40

Sample Output

1 7 19
1 19 37

分析题意：

• 将n个数依次按1，2，1，2，1，2… … 标号，然后删除下标为2的数
• 再将剩下的数依次按1，2，3，1，2，3… … 标号，然后删除下标为3的数
• 循环以上两步

样例1：
① 把1~20按照1，2，1，2，1，2… …排序，结果如图：

② 删除下标为2的数并按照1，2，3，1，2，3… … 排序，结果如图：

③ 删除下标为3的数并按照1，2，1，2，1，2…排序，结果如图：

④ 删除下标为2的数并按照1，2，3，1，2，3…排序，结果如图：

⑤ 删除下标为3的数并按照1，2，1，2…排序，结果如图：

此时人数不超过三人，故输出结果1 7 19
注意：

• erase( it )的作用是删除迭代器指向的当前元素，并返回该迭代器下一参数的迭代器。写法：it = li.erase(it);
• 注意最后一个元素后不能带空格，格式问题

AC代码

#include <cstdio>
#include <list>
#include <string>
#include <cstring>
#include <iostream>
#include <algorithm>using namespace std;int main()
{int t;//多组数据cin >> t;list<int> li;//STL:链表while (t--){li.clear();//清空上次操作的数据int n;cin >> n;for (int i = 1;i <= n;i++) li.push_back(i);//输入1~n个数int k = 2;list<int>::iterator it;while (li.size() > 3)//链表中的元素个数大于3{int num = 1;//num模拟链表中元素的下标1,2,1,2,1,2... ...for (it = li.begin();it != li.end();){if (num++ % k == 0){it = li.erase(it);//这句话执行以后会返回下一个迭代器}else it++;}//若以上for循环对2取模,则下一次应该对3取模;//若以上for循环对3取模,则下一次应该对2取模;//即k == 2? k = 3:k = 2;if (k == 2) k = 3;else k = 2;}//输出结果(注意最后一个元素后不能带空格)for (it = li.begin();it != li.end();it++) {if (it != li.begin()) cout << " ";cout << *it;}cout << endl;}return 0;
}

本题需要着重理解以下代码如何做到轮流对2和3取模，并且上轮删除对2取模为0的数，下轮删除对3取模为0的数

while (li.size() > 3){int num = 1;for (it = li.begin();it != li.end();){if (num++ % k == 0){it = li.erase(it);}else it++;}if (k == 2) k = 3;else k = 2;}