C++-小熊的果籃（CSP-J 2021 T4） 解題思路
阿新 • • 發佈：2022-05-28
【Horn Studio】程式設計專欄：小熊的果籃（CSP-J 2021 T4）解題思路
題目
題目描述
小熊的水果店裡擺放著一排 nn 個水果。每個水果只可能是蘋果或桔子，從左到右依次用正整數 1,2,…,n1,2,…,n 編號。連續排在一起的同一種水果稱為一個"塊"。小熊要把這一排水果挑到若干個果籃裡, 具體方法是： 每次都把每一個"塊"中最左邊的水果同時挑出，組成一個果籃。重複這一操作，直至水果用完。注意，每 次挑完一個果籃後, "塊"可能會發生變化。比如兩個蘋果“塊"之間的唯一桔子被挑走後, 兩個蘋果“塊"就變 成了一個"塊"。請幫小熊計算每個果籃裡包含的水果。輸入
第一行, 包含一個正整數第二行, 包含 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% 的資料,
對於 100%100% 的資料, 1≤n≤2×1051≤n≤2×105 。
來源
題目資訊
思路
因為每一次拿走水果都相當於是從序列裡刪除一個元素，那麼就可以用到連結串列。
再順著往下推，推出正解也就不遠了。
可以用一個 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; }