【Horn Studio】编程专栏：小熊的果篮（CSP-J 2021 T4）解题思路

题目

题目描述

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

输入

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

输出

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

样例输入 复制

20
1 1 1 1 0 0 0 1 1 1 0 0 1 0 1 1 0 0 0 0

样例输出 复制

1 5 8 11 13 14 15 17
2 6 9 12 16 18
3 7 10 19
4 20

提示

对于 10%10% 的数据, n≤5n≤5 。
对于 30%30% 的数据, n≤1000n≤1000 。
对于 70%70% 的数据, n≤50000n≤50000 。
对于 100%100% 的数据, 1≤n≤2×1051≤n≤2×105 。

来源

题目信息

• 提交: 12正确: 9
• 时间限制: 1 秒
• 内存限制: 512 MB
• 难度等级: 4 级
• 创建人: lwc
• 提交

 

思路

因为每一次拿走水果都相当于是从序列里删除一个元素，那么就可以用到链表。

再顺着往下推，推出正解也就不远了。

可以用一个 vector 容器（令其为 b）来存储每个块的块头，然后不断更新。具体如何更新呢？更新必须满足条件：去掉块头后这个块内仍然有元素。否则跳过（那块都消失了你还管人家干啥）。

就是这么简单。

&花絮&：哎，用数组直接模拟他不香吗………………90行代码WA了……………………太臭了！

代码

#include <bits/stdc++.h>

#define MAXN 200100

using namespace std;

int n, a[MAXN], l[MAXN], r[MAXN];
vector<int> b;

int main() {
    scanf("%d", &n);
    a[0] = a[n + 1] = -1, r[0] = 1, l[n + 1] = n;
    for (int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        if (a[i] != a[i - 1]) b.push_back(i);
        l[i] = i - 1, r[i] = i + 1;
    }
    while (r[0] != n + 1) {
        vector<int> tmp;
        for (int i = 0; i < b.size(); i++) {
            printf("%d ", b[i]);
            int u = l[b[i]], v = r[b[i]];
            r[u] = v, l[v] = u;
            if (a[b[i]] != a[u] && a[b[i]] == a[v]) tmp.push_back(v); 
        }
        puts("");
        b = tmp;
    }
    return 0;
}