ACM学习历程—HDU 1276 士兵队列训练问题(队列)

Description

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

Input

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

Output

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

Sample Input

2

20

40

Sample Output

1 7 19

1 19 37

题目大意就是一个队列报数，第一次12报数，报到2的出列；第二次123报数，报到3的出列；如此循环往复，直到队列人数不超过3.

这里采用两个队列交替使用，对于12报数的情况，只有报到1的才会入另一个队。

对于123的，自然是报到12的入另一个队。

代码：

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <set>
#include <stack>
#include <map>
#include <queue>
#include <string>
#include <algorithm>
#define LL long long

using namespace std;

int n;

void work()
{
    queue<int> q[2];
    bool now = 0;
    int num;

    for (int i = 1; i <= n; ++i)
        q[now].push(i);

    for (;;)
    {
        if (q[now].size() <= 3)
            break;
        num = 0;
        now = !now;
        while (!q[!now].empty())
        {
            num = num%2+1;
            if (num != 2)
                q[now].push(q[!now].front());
            q[!now].pop();
        }

        if (q[now].size() <= 3)
            break;
        num = 0;
        now = !now;
        while (!q[!now].empty())
        {
            num = num%3+1;
            if (num != 3)
                q[now].push(q[!now].front());
            q[!now].pop();
        }
    }
    bool flag = false;
    while (!q[now].empty())
    {
        if (flag)
            printf(" ");
        printf("%d", q[now].front());
        q[now].pop();
        flag = true;
    }
    printf("\n");
}

int main()
{
    //freopen("test.in", "r", stdin);
    int T;
    scanf("%d", &T);
    for (int times = 0; times < T; ++times)
    {
        scanf("%d", &n);
        work();
    }
    return 0;
}