小熊的果篮 2021 CSP J2

小熊的果篮

题目描述

小熊的水果店里摆放着一排 n个水果。每个水果只可能是苹果或桔子，从左到右依次用正整数 1、2、3、……、n编号。
连续排在一起的同一种水果称为一个“块”。
小熊要把这一排水果挑到若干个果篮里，具体方法是：每次都把每一个“块”中最左边的水果同时挑出，组成一个果篮。重复这一操作，直至水果用完。
注意，每次挑完一个果篮后，“块”可能会发生变化。
比如两个苹果“块”之间的唯一桔子被挑走后，两个苹果“块”就变成了一个“块”。
请帮小熊计算每个果篮里包含的水果。
输入格式
输入的第一行包含一个正整数 n，表示水果的数量。
输入的第二行包含 n个空格分隔的整数，其中第 i 个数表示编号为 i 的水果的种类，1 代表苹果，0代表桔子。
输出格式
输出若干行。
第 i行表示第 i次挑出的水果组成的果篮。
从小到大排序输出该果篮中所有水果的编号，每两个编号之间用一个空格分隔。
数据范围
对于 10%的数据，n≤5。
对于 30% 的数据，n≤1000。
对于 70% 的数据，n≤50000。
对于 100% 的数据，n≤2×10⁵。
输入样例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
样例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的水果被挑了出来。
输入样例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

解析

直接用两个队列，暴力模拟，但时间复杂度较高。
可以考虑将类型相同的连续水果看成一个整体，定义成块block。队列中存放的元素是块block， 所有的进队，出队操作都是基于块block进行的。此外，本来互相独立的块，由于中间分隔它们的唯一水果被挑走后，两块就合成了一块，所以需要实现合并块的算法代码。
同时也需要开一个布尔数组，用于标记哪些水果已经被挑选了出来。

#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;

struct block{ // block 
	int st, ed, obn;//st 块首，ed 块尾，obn 类型
}f, x, ZJ;
int n, cnt, sob[200005];
queue<block> q, q2;
bool used[200005]; 

int main(){
	scanf("%d", &n);
	for (int i = 1;i <= n;i ++) scanf("%d", &sob[i]);
	sob[n + 1] = !sob[n];//构建右侧边界
	for (int i = 2, s = 1;i <= n + 1;i ++){
		if (sob[i] != sob[i - 1]) 
		{
			q.push((block){s, i - 1, sob[i - 1]});
		    s = i; 
		}//make BLOCK
	}
	cnt = n;//水果总个数
	while (cnt){
		if(q.size()==1)//optimized due to only one block left
		{
			f = q.front();
			q.pop();
			for(int j=f.st;j<=f.ed;j++)
			{
			  	if(!used[j]) {
			  		printf("%d\n", j), cnt --;//这里可以不用自减
				}
			}
			break;
		} 
		while (q.size()){
			f = q.front();//取第一块
			q.pop();//删除第一块
			while (used[f.st] && f.st < f.ed) f.st ++; 
			printf("%d ", f.st), cnt --;
			used[f.st] = 1; //MARK already output
			if (f.ed == f.st) continue; //不必再次入队，说明此块已被删除
			f.st ++;
			q2.push(f); //改变st后，进队备用
		}
		puts("");
		while (q2.size()){
			ZJ = q2.front();
			q2.pop();
			while (q2.size()){
				x = q2.front();
				if (x.obn == ZJ.obn){ //MERGE
					ZJ.ed = x.ed;
					q2.pop();
				}
				else break;
			}
			q.push(ZJ); // PUT INTO orgin q
		}
	}
	return 0;
}