问题 D: 2.4.4 士兵队列训练

题目描述

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

输入格式

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

输出格式

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

输入样例 复制

2 
20 
40

输出样例 复制

1 7 19
1 19 37

完整代码

#include<bits/stdc++.h>
using namespace std;

int main(){
    //freopen("/config/workspace/test/test","r",stdin);
    int n;
    cin>>n;
    while(n--){
        int t;
        cin>>t;
        t++;
        int a[t];
        for(int i=1;i<t;i++){
            a[i]=i;//下标是最初的编号
        }
        int num=t-1;
        while(num>3){
            if(num>3){
                int count=0;
                for(int i=1;i<t;i++){
                    if(a[i]%2==0&&a[i]!=0){//报2的出列
                        a[i]=0;
                        num--;
                    }else if(a[i]!=0){
                        a[i]=++count;//count是新编号，i是初始编号
                    }
                }
            }
            if(num>3){
                int count=0;
                for(int i=1;i<t;i++){
                    if(a[i]%3==0&&a[i]!=0){//报3的出列
                        a[i]=0;
                        num--;
                    }else if(a[i]!=0){
                        a[i]=++count;//count是新编号，i是初始编号
                    }
                }
            }
        }
        for(int i=1;i<t;i++){
            if(a[i]!=0) cout<<i<<" ";
        }
        cout<<endl;

    }


    return 0;
}