当前位置:首页 > 数据结构 > 链表 > 正文
洛谷P7912小熊的果篮(CSPJ2021)
259+

题目大意:n个由0和1组成的序列,连续相同的数字为一块,依次输出每一块数字左边的编号并删除这个元素,直到序列为空。

题目描述

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

输入输出格式

输入格式

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

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

输出格式

输出若干行。

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

输入输出样例

输入样例 #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

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

见附件中的 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` 语句输入、输出。

解题思路

第一次输出的是0还是1,取决于最左边的数字是0还是1;输出0后,需要在右边找第一个出现的1;输出1后,需要找右边第一个出现的0……一趟一趟解决!

怎么找右边最近的0或者1?将0和1分开两个数组存储,下标具有单调性即可二分查找,可以用set维护;当然,使用双向链表,可以再开个数组,记录每一块的开头位置在哪,边删除边维护即可。

程序实现

双向链表做法:删除元素,容易想到链表O(1)操作,只是链表不支持随机访问,这时我们可以开个数组,记录每一块的开头,每次输出块头后,删除该元素,如果下一个元素跟块头相同,那么这一块还没删完,如果还跟上一个元素、上一块不同,那么这一块需要保留下来。(感谢shenghang16提供思路)

set维护:

暴力骗100分:

洛谷P7912小熊的果篮(CSPJ2021):等您坐沙发呢!

发表评论

😉😐😡😈🙂😯🙁🙄😛😳😮mrgreen.png😆💡😀👿😥😎😕

快捷键:Ctrl+Enter