小熊的果篮

提交数: 256, 通过率: 43.75%, 平均分: 67.19

题目描述:

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

输入格式:

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

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

输出格式:

输出若干行。

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

样例输入:

样例1
12
1 1 0 0 1 1 1 0 1 1 0 0

样例2
20
1 1 1 1 0 0 0 1 1 1 0 0 1 0 1 1 0 0 0 0

样例输出:

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

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

提示:

【样例解释 #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≤5。
对于 30% 的数据,n≤1000。
对于 70% 的数据,n≤50000。
对于 100% 的数据,1≤n≤2×105

时间限制: 1000ms
空间限制: 512MB

来源: CSP2021普及T4