洛谷P7912 小熊的果篮

题目描述

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

输入格式

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

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

输出格式

输出若干行。

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

——————————————————分割线——————————————————————

我的代码（70分，后三个测试点TLE了），用数组模拟

#include<bits/stdc++.h>
using namespace std;

int n,nn;
int const M=1e5+10,N=2*M;
int s[N];
bool st[N];

void gg(){
    int sum=0;
    for(int i=1;i<=nn;i++)
        if(st[i]) {
            printf("%d ",i);
            st[i]=false;
            sum=s[i];
            n--;
            break;
        }
    for(int i=2;i<=nn;i++){
        if(st[i]&&s[i]!=sum){
            printf("%d ",i);
            st[i]=false;
            sum=s[i];
            n--;
        }
    }
    cout<<endl;
return;
}

int main(){
    scanf("%d",&n);
    nn=n;
    for(int i=1;i<=nn;i++) scanf("%d",&s[i]);
    for(int i=1;i<=nn;i++) st[i]=true;
    while(n>0) gg();
return 0;
}

粘贴一个洛谷YuJiahe大佬的题解：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <queue>

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 里
		}
	}
}