P7912 [CSP-J 2021] 小熊的果篮

# [CSP-J 2021] 小熊的果篮

## 题目描述

小熊的水果店里摆放着一排 $n$ 个水果。每个水果只可能是苹果或桔子，从左到右依次用正整数 $1, 2, \ldots, n$ 编号。连续排在一起的同一种水果称为一个“块”。小熊要把这一排水果挑到若干个果篮里，具体方法是：每次都把每一个“块”中最左边的水果同时挑出，组成一个果篮。重复这一操作，直至水果用完。注意，每次挑完一个果篮后，“块”可能会发生变化。比如两个苹果“块”之间的唯一桔子被挑走后，两个苹果“块”就变成了一个“块”。请帮小熊计算每个果篮里包含的水果。

## 输入格式

第一行，包含一个正整数 $n$，表示水果的数量。

第二行，包含 $n$ 个空格分隔的整数，其中第 $i$ 个数表示编号为 $i$ 的水果的种类，$1$ 代表苹果，$0$ 代表桔子。

## 输出格式

输出若干行。

第 $i$ 行表示第 $i$ 次挑出的水果组成的果篮。从小到大排序输出该果篮中所有水果的编号，每两个编号之间用一个空格分隔。

## 样例 #1

### 样例输入 #1

```
12
1 1 0 0 1 1 1 0 1 1 0 0
```

### 样例输出 #1

```
1 3 5 8 9 11
2 4 6 12
7
10
```

## 样例 #2

### 样例输入 #2

```
20
1 1 1 1 0 0 0 1 1 1 0 0 1 0 1 1 0 0 0 0
```

### 样例输出 #2

```
1 5 8 11 13 14 15 17
2 6 9 12 16 18
3 7 10 19
4 20
```

## 样例 #3

### 样例输入 #3

```
见附件中的 fruit/fruit3.in。
```

### 样例输出 #3

```
见附件中的 fruit/fruit3.ans。
```

## 提示

**【样例解释 #1】**

这是第一组数据的样例说明。

所有水果一开始的情况是 $[1, 1, 0, 0, 1, 1, 1, 0, 1, 1, 0, 0]$，一共有 $6$ 个块。

在第一次挑水果组成果篮的过程中，编号为 $1, 3, 5, 8, 9, 11$ 的水果被挑了出来。

之后剩下的水果是 $[1, 0, 1, 1, 1, 0]$，一共 $4$ 个块。

在第二次挑水果组成果篮的过程中，编号为 $2, 4, 6, 12$ 的水果被挑了出来。

之后剩下的水果是 $[1, 1]$，只有 $1$ 个块。

在第三次挑水果组成果篮的过程中，编号为 $7$ 的水果被挑了出来。

最后剩下的水果是 $[1]$，只有 $1$ 个块。

在第四次挑水果组成果篮的过程中，编号为 $10$ 的水果被挑了出来。

**【数据范围】**

对于 $10 \%$ 的数据，$n \le 5$。
对于 $30 \%$ 的数据，$n \le 1000$。
对于 $70 \%$ 的数据，$n \le 50000$。
对于 $100 \%$ 的数据，$1 \le n \le 2 \times {10}^5$。

**【提示】**

由于数据规模较大，建议 C/C++ 选手使用 `scanf` 和 `printf` 语句输入、输出。

【算法分析】

模拟。

【AC代码】

#include<stdio.h>
#include<iostream>
#include<queue>
using namespace std;
const int N=2e5+10;
inline int fread()
{char ch=getchar();int n=0,m=1;while(ch<'0' or ch>'9'){if(ch=='-')m=-1;ch=getchar();}while(ch>='0' and ch<='9')n=(n<<3)+(n<<1)+ch-48,ch=getchar();return n*m;
}
int n,sum,a[N];
bool flag[N];
struct node
{int m,x,y;
}z,p,q;
queue<node> q1,q2;
signed main()
{n=fread();for(int i=1;i<=n;i++)a[i]=fread();a[n+1]=!a[n];for(int i=2,j=1;i<n+2;i++)if(a[i]!=a[i-1])q1.push((node){j,i-1,a[i-1]}),j=i;//把连续的相同水果合并成一块sum=n;while(sum){while(q1.size()){z=q1.front(),q1.pop();while(flag[z.m] and z.m<=z.x)z.m++;//如果被取了if(z.m>z.x)continue;printf("%d ",z.m),sum--,flag[z.m]=true;//将块的第一个水果取出并输出if(z.x==z.m)continue;//块被取完了z.m++,q2.push(z);//存到q2里合并}printf("\n");while(q2.size()){q=q2.front(),q2.pop();while(q2.size()){p=q2.front();if(p.y==q.y)q.x=p.x,q2.pop();//合并else break;}q1.push(q);//放回q1}}return 0;
}

我橙名了！！！