HDOJ 1276 士兵队列训练问题

士兵队列训练问题

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

Problem Description

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

 

Input

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

 

Output

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

 

Sample Input

  
  

   
   2
20
40
  
  

 

Sample Output

  
  

   
   1 7 19
1 19 37
  
  

 

    注意是必须每一轮都点完，而不是一数到只剩三人了就停止

#include <cstdio>
using namespace std;

int main()
{
    int t;
    int n;
    int a[5000 + 10];
    scanf("%d", &t);
    while (t--){
        scanf("%d", &n);
        int sum = n;
        for (int i = 1; i <= n; i ++)
            a[i] = i;

        if (n <= 3){
            printf("1");
            for (int i = 2; i <= n; i++){
                if (a[i])
                    printf(" %d", a[i]);
            }
            printf("\n");
            continue;
        }
        while (1){
            int flag = 0;
            for (int i = 1; i <= n; i++){
                if (a[i])
                    flag++;
                if (flag == 2){
                    a[i] = 0;
                    flag = 0;
                    sum--;
                }
            }
            if (sum <= 3)
                break;

            flag = 0;
            for (int i = 1; i <= n; i++){
                if (a[i])
                    flag++;
                if (flag == 3){
                    a[i] = 0;
                    flag = 0;
                    sum--;
                }
            }
            if (sum <= 3)
                break;
        }
        printf("1");
        sum--;
        for (int i = 2; sum; i++){
            if (a[i]){
                printf(" %d", a[i]);
                sum--;
            }
        }
        printf("\n");
    }
    return 0;
}