题目大意:n个由0和1组成的序列,连续相同的数字为一块,依次输出每一块数字左边的编号并删除这个元素,直到序列为空。
题目描述
输入输出格式
输入格式
第一行,包含一个正整数 $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分: