小熊的果篮(fruit)【CSPJ2021】

题目描述

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

输入输出格式

输入格式：

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

输出格式：

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

输入输出样例

输入样例#1：

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

提示信息

对于 100% 的数据，n ≤ 2 × 10^5。

题目分析

这道题没有算法，不，是模拟算法，根据题目的意思做：

1.定义一个“块”：

用结构体来做更好：

struct kuai
{int st;int ed;int val;
};

2.存入块中：

    cin>>n;for(int i=1;i<=n;i++) cin>>d[i];if(d[n]==0) d[n+1]=1;else d[n+1]=0;for(int i=2;i<=n+1;i++){if(d[i]!=d[i-1]) {q.push((kuai){tempst,i-1,d[i-1]});tempst=i;}}

3.取每个块最左边的水果：

从原块开始处往右走，遇到没有标记的就取，然后标记

        while(!q.empty()){kuai temp=q.front();q.pop();while(vis[temp.st]&&temp.st<=temp.ed) temp.st++;if(temp.st>temp.ed) continue;cout<<temp.st<<" ";cnt--;vis[temp.st]=1;if(temp.ed==temp.st) continue;temp.st++;q2.push(temp);}

4.合并相邻两“块”：

        while(!q2.empty()){kuai temp=q2.front();q2.pop();while(!q2.empty()){kuai temp2=q2.front();if(temp2.val==temp.val)//同一种水果{temp.ed=temp2.ed;q2.pop();}else break;}q.push(temp);}

完整代码：

#include<bits/stdc++.h>
using namespace std;
struct kuai
{int st;int ed;int val;
};
int n,d[200002],tempst=1;
queue<kuai> q,q2;
bool vis[200001];
int main()
{
//  freopen("fruit.in","r",stdin);
//  freopen("fruit.out","w",stdout);cin>>n;for(int i=1;i<=n;i++) cin>>d[i];if(d[n]==0) d[n+1]=1;else d[n+1]=0;for(int i=2;i<=n+1;i++){if(d[i]!=d[i-1]) {q.push((kuai){tempst,i-1,d[i-1]});tempst=i;}}int cnt=n;while(cnt){while(!q.empty()){kuai temp=q.front();q.pop();while(vis[temp.st]&&temp.st<=temp.ed) temp.st++;if(temp.st>temp.ed) continue;cout<<temp.st<<" ";cnt--;vis[temp.st]=1;if(temp.ed==temp.st)continue;temp.st++;q2.push(temp);}cout<<endl;while(!q2.empty()){kuai temp=q2.front();q2.pop();while(!q2.empty()){kuai temp2=q2.front();if(temp2.val==temp.val){temp.ed=temp2.ed;q2.pop();}else break;}q.push(temp);}}return 0;
}