数据结构-第四次作业-behind

数据结构与算法实验题Who is behind
★实验任务
开学了，有N 个新生需要排成一列。这N 个新生编号从1 到N，为了排队方便，按编号
从1 到N 顺序入队伍，起始队伍为空，每个人进队伍的选择有两种，一种是直接排在当前队
伍的最前面，一种是排在当前队伍最末端。现在想知道所有人都入队伍之后每人的后面是谁。
★数据输入
输入第一行为一个正整数N (2 < N < 10000)， 表示有N 个人，编号为1 到N。。
接下来1 行，有N 个整数，0 或者1。第i 个数表示第i 个同学是排在当前队伍最前端
还是在最末端。（0 表示排在当前队伍最前端，1 表示排在当前队伍最末端）
★数据输出
输出N 行每行一个整数。第i 行的整数表示全部人都入队伍后第i 个同学后面的同学的
编号，如果后面没人的同学则输出-1。
输入示例输出示例
6010101

24163-1

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
struct node
{
	int x,y;
}q[10005];
bool cmp(node a,node b)
{
	return a.y<b.y;
}
int main()
{
	int i,j,a,b,n,t;
	scanf("%d",&n);
	a=b=0;
	for(i=1;i<=n;i++)
	{
		
		
		scanf("%d",&t);
		if(t==0)
		{
			a--;
			q[i].x=i;
			q[i].y=a;//有0时就--；是让他进入到队伍的最前端。
		}
		else
		{
			b++;
			q[i].x=i;
			q[i].y=b;//同理。队伍的最尾端。
			
			
		}
		
	}

	sort(q+1,q+n+1,cmp);//这里先进行排序；
	for(i=1;i<=n;i++)
	{
		for(j=1;j<=n;j++)
		{
		if (q[j].x==i)		//只要找到在新排序的队列中，q[y].x=n;输出q[y+1].x也就是他后面的数字就行。
		{
			if(j==n)
				printf("-1\n");
			else
			printf("%d\n",q[j+1].x);
			break;
		}
		}
	}
	return 0;
	
	
	
	
}