CSPJ2021D. 小熊的果篮 (Bear‘s Fruit Basket)

题目描述

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

输入格式

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

输出格式

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

输入数据 1

12
1 1 0 0 1 1 1 0 1 1 0 0

Copy

输出数据 1

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

Copy

样例 1 解释

这是第一组数据的样例说明。
所有水果一开始的情况是 $110011101100$，一共有 $6$ 个块。
在第一次挑水果组成果篮的过程中，编号为 $1358911$ 的水果被挑了出来。
之后剩下的水果是 $101110$，一共 $4$ 个块。
在第二次挑水果组成果篮的过程中，编号为 $24612$ 的水果被挑了出来。
之后剩下的水果是 $11$，只有 $1$ 个块。
在第三次挑水果组成果篮的过程中，编号为 $7$ 的水果被挑了出来。
最后剩下的水果是 $1$，只有 $1$ 个块。
在第四次挑水果组成果篮的过程中，编号为 $10$ 的水果被挑了出来。

输入数据 2

20
1 1 1 1 0 0 0 1 1 1 0 0 1 0 1 1 0 0 0 0

Copy

输出数据 2

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

Copy

样例 3

见选手目录下的 fruit3.in 与 fruit3.ans。

数据范围与提示

对于 $10\%$ 的数据，$n\le 5$。
对于 $30\%$ 的数据，$n\le 1000$。
对于 $70\%$ 的数据，$n\le 50000$。
对于 $100\%$ 的数据，n≤2×105n \le 2 \times 10^5n≤2×105。
由于数据规模较大，建议 C/C++ 选手使用 scanf 和 printf 语句输入、输出。

#include 
#include 
#include 
#include 
#include 

using namespace std;

struct kuai{ // 块
	int st, ed, th;
}f, x, ad;
int n, cnt, t[200002];
queue<kuai> q, q2;
bool used[200001]; // 记录是否被取出
int main(){
	scanf("%d", &n);
	for (int i = 1;i <= n;i ++) scanf("%d", &t[i]);
	t[n + 1] = !t[n];
	for (int i = 2, si = 1;i <= n + 1;i ++){
		if (t[i] != t[i - 1]) q.push((kuai){si, i - 1, t[i - 1]}), si = i; // 把连续一段相同的元素合成一个块
	}
	cnt = n;
	while (cnt){
		while (q.size()){
			f = q.front();
			q.pop();
			while (used[f.st] && f.st <= f.ed) f.st ++; // 如果已经被取了
			if (f.st > f.ed) continue;
			printf("%d ", f.st), cnt --;
			used[f.st] = 1; // 将块的开头元素去掉并输出
			if (f.ed == f.st) continue; // 如果这个块被取完了
			f.st ++;
			q2.push(f); // 先临时存到 q2 里进行合并
		}
		putchar('\n');
		while (q2.size()){
			ad = q2.front();
			q2.pop();
			while (q2.size()){
				x = q2.front();
				if (x.th == ad.th){ // 能合并就合并
					ad.ed = x.ed;
					q2.pop();
				}
				else break;
			}
			q.push(ad); // 丢回 q 里
		}
	}
}