社区讨论
80TLE#8#9,求优化
P7912[CSP-J 2021] 小熊的果篮参与者 1已保存回复 1
讨论操作
快速查看讨论及其快照的属性,并进行相关操作。
- 当前回复
- 1 条
- 当前快照
- 1 份
- 快照标识符
- @lwvgo1p2
- 此快照首次捕获于
- 2024/06/01 09:55 2 年前
- 此快照最后确认于
- 2024/06/01 12:06 2 年前
CPP
#include <iostream>
#include <list>
#include <vector>
using namespace std;
int main() {
int n;
cin >> n;
vector<int> fruits(n);
for (int i = 0; i < n; ++i) {
cin >> fruits[i];
}
list<pair<int, int>> blocks; // Pair of (start_index, fruit_type)
int start = 0;
// Construct initial blocks
for (int i = 1; i <= n; ++i) {
if (i == n || fruits[i] != fruits[i - 1]) {
blocks.emplace_back(start, fruits[start]);
start = i;
}
}
while (!blocks.empty()) {
vector<int> basket;
list<pair<int, int>>::iterator it = blocks.begin();
int prev_type = -1;
while (it != blocks.end()) {
if (it->second != prev_type) {
basket.push_back(it->first + 1); // Store 1-based index
++(it->first);
prev_type = it->second;
}
if (it->first == n || fruits[it->first] != it->second) {
it = blocks.erase(it);
} else {
++it;
}
}
for (const int& index : basket) {
cout << index << " ";
}
cout << endl;
}
return 0;
}
回复
共 1 条回复,欢迎继续交流。
正在加载回复...