社区讨论

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 条回复,欢迎继续交流。

正在加载回复...