hdu 1276 士兵队列训练问题

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;

int main()
{
    int n,c,flag,j,tmp;
    bool data[5005];
    cin>>c;
    while(c--)
    {
        cin>>n;
        tmp=n;
        flag=1;
        memset(data,true,sizeof(data));
        while(tmp>3)
        {
            if(flag%2!=0)
            {
                 j=0;
                 for(int i=1;i<=n;i++)
                 {
                    if(data[i]==true)
                    j++;
                    if(j%2==0&&data[i]==true)
                    {
                      data[i]=false;
                      tmp--;
                    }
                 }
                 flag++;
            }
            else
            {
                j=0;
                for(int i=1;i<=n;i++)
                {
                    if(data[i]==true)
                    j++;
                    if(j%3==0&&data[i]==true)
                    {
                        data[i]=false;
                         tmp--;
                    }
                }
                flag++;
            }
        }
        int ji=0;
        int a[3];
        for(int i=1;i<=n;i++)
        {
            if(data[i]==true)
            {
                a[ji]=i;
                ji++;
            }
        }
        for(int i=0;i<ji-1;i++)
        {
            cout<<a[i]<<" ";
        }
        cout<<a[ji-1]<<endl;
    }
    return 0;
}