士兵队列训练问题

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 8914 Accepted Submission(s): 3940

Problem Description

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

Input

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

Output

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

Sample Input

2 20 40

Sample Output

1 7 19 1 19 37

一道简单的模拟题，被一个括号坑了好久，重写好几遍脑子都晕了，还是没找到错误，后面突然发现了问题。。。少了个花括号

AC代码：

# include <stdio.h>
# include <string.h>
# include <stdlib.h>
# include <iostream>
# include <fstream>
# include <vector>
# include <queue>
# include <stack>
# include <map>
# include <math.h>
# include <algorithm>
using namespace std;
# define pi acos(-1.0)
# define mem(a,b) memset(a,b,sizeof(a))
# define FOR(i,a,n) for(int i=a; i<=n; ++i)
# define For(i,n,a) for(int i=n; i>=a; --i)
# define FO(i,a,n) for(int i=a; i<n; ++i)
# define Fo(i,n,a) for(int i=n; i>a ;--i)
typedef long long LL;
typedef unsigned long long ULL;

int a[5005];

int main()
{
    int t,n;
    cin>>t;
    while(t--)
    {
        cin>>n;
        for(int i=1; i<=n; i++)a[i]=1;
        if(n<=3)
        {
            if(n==1)cout<<1<<endl;
            else if(n==2)cout<<"1 2"<<endl;
            else if(n==3)cout<<"1 2 3"<<endl;
        }
        else
        {
            int sum=n;
            int flag=1;
            while(sum>3)
            {
                int s=0;
                if(flag%2)///报2的出列
                {
                    for(int i=1; i<=n; i++)
                    {
                        if(a[i])
                        {
                            s++;
                            if(s%2==0)a[i]=0,sum--;
                        }
                    }
                    flag=!flag;
                }
                else///报3的出列
                {
                    for(int i=1; i<=n; i++)
                    {
                        if(a[i])
                        {
                            s++;
                            if(s%3==0)a[i]=0,sum--;
                        }
                    }
                    flag=!flag;
                }
            }
            for(int i=1; i<=n; i++)
                if(a[i]==1)printf(i==1?"%d":" %d",i);
            cout<<endl;
        }
    }
    return 0;
}