Submission #397578

#	Time	Username	Problem	Language	Result	Execution time	Memory
397578		timmyfeng	Rope (JOI17_rope)	C++17	80 / 100	2591 ms	116708 KiB

rope

#include <bits/stdc++.h>
using namespace std;

const int N = 1000001;

int color[N], tally[N], ans[N];
vector<int> pairs[N];

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, m;
    cin >> n >> m;

    for (int i = 0; i < n; ++i) {
        cin >> color[i];
        ++tally[color[i]];
    }

    multiset<int> maxi = {0};
    for (int i = 1; i <= m; ++i) {
        maxi.insert(tally[i]);
        ans[i] = INT_MAX;
    }

    for (int k = 0; k < 2; ++k) {
        fill(pairs, pairs + m + 1, vector<int>());
        for (int i = k; i + 2 <= n; i += 2) {
            if (color[i] != color[i + 1]) {
                pairs[color[i]].push_back(color[i + 1]);
                pairs[color[i + 1]].push_back(color[i]);
            }
        }

        for (int i = 1; i <= m; ++i) {
            maxi.erase(maxi.find(tally[i]));

            sort(pairs[i].begin(), pairs[i].end());

            for (int j = 0, k = 0; j < (int) pairs[i].size(); j = k) {
                while (k < (int) pairs[i].size() && pairs[i][j] == pairs[i][k]) {
                    ++k;
                }
                maxi.erase(maxi.find(tally[pairs[i][j]]));
                maxi.insert(tally[pairs[i][j]] - (k - j));
            }

            ans[i] = min(ans[i], n - tally[i] - *(--maxi.end()));

            for (int j = 0, k = 0; j < (int) pairs[i].size(); j = k) {
                while (k < (int) pairs[i].size() && pairs[i][j] == pairs[i][k]) {
                    ++k;
                }
                maxi.erase(maxi.find(tally[pairs[i][j]] - (k - j)));
                maxi.insert(tally[pairs[i][j]]);
            }

            maxi.insert(tally[i]);
        }
    }

    for (int i = 1; i <= m; ++i) {
        cout << ans[i] << "\n";
    }
}

• Subtask 1: 15 / 15
• Subtask 2: 30 / 30
• Subtask 3: 10 / 10
• Subtask 4: 25 / 25
• Subtask 5: 0 / 20