杭电ACM 1276 士兵队列训练

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<malloc.h>
//定义队列结构体:
typedef struct str
{
 int capacity;  //定义队列的容量
 int size;    //定义队列的长度
 int head;    //定义队列的头
 int tail;    //定义队列的尾
 int *array;   //定义队列数组，存放队列的数据
} queue;//typedef函数，用queue 代替struct str

//定义此函数使每次队列用过之后可以重新赋0
void fqueue(queue *p)
{
 p->tail=0;
 p->head=0;
 p->size=0;
}

//定义此函数使函数使用之前赋值0，并确定队列的最大容量
void gqueue(queue *p,int n)
{
 p->capacity=n;
 p->tail=0;
 p->head=0;
 p->size=0;
 p->array=(int *)malloc(sizeof(queue)*n);//注意给数组分配空间
}

//向队列中输入数据，进栈
int getqueue(queue *p,int num)
{
 p->array[p->tail]=num;
 p->tail++;
 p->size++;
 return 0;
}

//从队列中输出数据,出栈
int outqueue(queue *p)
{
 int num;
 num=p->array[p->head];
 p->array[p->head]=0;
 p->head++;
 p->size++;
 return num;//注意此处是return num;
}

//释放空间
void freequeue(queue *p)
{
 free(p);
}

//主函数部分
int main()
{
 int cases,i,a,num;
 queue *p1=NULL,*p2=NULL,*ans=NULL;//定义三个队列，ans用来存放要输出队列的值
// p1=NULL;
// p2=NULL;    （这样赋值也可以）
// ans=NULL;
 scanf("%d",&cases);
 while(cases--){
  scanf("%d",&a);
 // gqueue(p1,a);  这两句放在此处不正确，因为要先给p1,p2分配空间才可以确定其最大容量
 // gqueue(p2,a);
     p1=(queue *)malloc(sizeof(queue));
        p2=(queue *)malloc(sizeof(queue)); //分配空间是注意p1,p2是queue类型的
  gqueue(p1,a);
  gqueue(p2,a);
  for(i=1;i<=a;i++)
  getqueue(p1,i);
  ans=p1;
  while(p1->size>3){  //总循环条件
   fqueue(p2);
            while(p1->head!=p1->tail){  //当队列不为空时进行循环
              num=outqueue(p1);
   if((p1->head)%2!=0)
    getqueue(p2,num);
  }
   if(p2->size<=3){  //注意每次交换队列中的数据后要判断是否<=3
   ans=p2;
   break;
  }
  
     fqueue(p1);
     while(p2->head!=p2->tail){   //当队列不为空时进行循环
             num=outqueue(p2);
      if((p2->head)%3!=0)
     getqueue(p1,num);
     }
     if(p1->size<=3){  //注意每次交换队列中的数据后要判断是否<=3
      ans=p1;
      break;
     }
  }

  //输出数据
     for(i=ans->head;i<ans->tail-1;i++)
      printf("%d ",ans->array[i]);
     printf("%d\n",ans->array[i]);
   freequeue(p1);
   freequeue(p2);
 }
 return 0;
}