当前位置：首页 > news >正文

打卡信奥刷题（3169）用C++实现信奥题 P7912 [CSP-J 2021] 小熊的果篮

news 2026/4/26 19:04:03

P7912 [CSP-J 2021] 小熊的果篮

题目描述

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

输入格式

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

第二行，包含nnn个空格分隔的整数，其中第iii个数表示编号为iii的水果的种类，111代表苹果，000代表桔子。

输出格式

输出若干行。

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

输入输出样例 #1

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

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

输入 #3

见附件中的 fruit/fruit3.in。

输出 #3

见附件中的 fruit/fruit3.ans。

说明/提示

【样例解释 #1】

这是第一组数据的样例说明。

所有水果一开始的情况是[1,1,0,0,1,1,1,0,1,1,0,0][1, 1, 0, 0, 1, 1, 1, 0, 1, 1, 0, 0][1,1,0,0,1,1,1,0,1,1,0,0]，一共有666个块。

在第一次挑水果组成果篮的过程中，编号为1,3,5,8,9,111, 3, 5, 8, 9, 111,3,5,8,9,11的水果被挑了出来。

之后剩下的水果是[1,0,1,1,1,0][1, 0, 1, 1, 1, 0][1,0,1,1,1,0]，一共444个块。

在第二次挑水果组成果篮的过程中，编号为2,4,6,122, 4, 6, 122,4,6,12的水果被挑了出来。

之后剩下的水果是[1,1][1, 1][1,1]，只有111个块。

在第三次挑水果组成果篮的过程中，编号为777的水果被挑了出来。

最后剩下的水果是[1][1][1]，只有111个块。

在第四次挑水果组成果篮的过程中，编号为101010的水果被挑了出来。

【数据范围】

对于10%10 \%10%的数据，n≤5n \le 5n≤5。
对于30%30 \%30%的数据，n≤1000n \le 1000n≤1000。
对于70%70 \%70%的数据，n≤50000n \le 50000n≤50000。
对于100%100 \%100%的数据，1≤n≤2×1051 \le n \le 2 \times {10}^51≤n≤2×105。

【提示】

由于数据规模较大，建议 C/C++ 选手使用scanf和printf语句输入、输出。

C++实现

#include<bits/stdc++.h>#defineINF200010usingnamespacestd;intn;set<int>s1,s2;//把下标扔在两个set里intmain(){scanf("%d",&n);s1.clear();s2.clear();intq;for(inti=1;i<=n;++i){scanf("%d",&q);if(q)s1.insert(i);elses2.insert(i);}s1.insert(INF);s2.insert(INF);intnw=0;boolp=*s1.begin()<*s2.begin()?0:1;while(!p&&s1.size()>1||p&&s2.size()>1)//因为有个INF，所以size要>1if(!p){nw=*s1.upper_bound(nw);//upper_bound找下一个if(nw==INF){nw=0;p=*s1.begin()<*s2.begin()?0:1;//p要重新选puts("");continue;}printf("%d ",nw);//取了，输出，删除s1.erase(nw);p=!p;//取另外一坨}else{nw=*s2.upper_bound(nw);if(nw==INF){nw=0;p=*s1.begin()<*s2.begin()?0:1;puts("");continue;}printf("%d ",nw);s2.erase(nw);p=!p;}puts("");while(s1.size()>1)printf("%d\n",*s1.begin()),s1.erase(*s1.begin());while(s2.size()>1)printf("%d\n",*s2.begin()),s2.erase(*s2.begin());return0;}