Submission #854345

#	Time	Username	Problem	Language	Result	Execution time	Memory
854345		mzh	Bob (COCI14_bob)	C++17	12 / 120	106 ms	18708 KiB

bob

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

using ll = long long;

#define all(x) begin(x), end(x)

int nxt() {
    int x;
    cin >> x;
    return x;
}

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

    int n = nxt(), m = nxt();
    vector<vector<int>> a(n, vector<int>(m));
    for (int i = 0; i < n; i++) {
        generate(all(a[i]), nxt);
    }

    vector<vector<int>> streak(n, vector<int>(m, 1));
    for (int i = 0; i < n; i++) {
        for (int j = 1; j < m; j++) {
            if (a[i][j] == a[i][j - 1]) {
                streak[i][j] += streak[i][j - 1];
            }
        }
    }

    ll ans = 0;
    vector<priority_queue<int>> lengths(m);
    vector<int> sum(m);
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if (i > 0 && a[i][j] != a[i - 1][j]) {
                lengths[j] = priority_queue<int>();
                sum[j] = 0;
            }
            while (!lengths[j].empty() && lengths[j].top() > streak[i][j]) {
                sum[j] -= lengths[j].top() - streak[i][j];
                lengths[j].pop();
            }
            lengths[j].push(streak[i][j]);
            sum[j] += streak[i][j];

            ans += sum[j];
        }
    }

    cout << ans << '\n';
}

• Subtask 1: 0 / 12
• Subtask 2: 12 / 12
• Subtask 3: 0 / 12
• Subtask 4: 0 / 12
• Subtask 5: 0 / 12
• Subtask 6: 0 / 12
• Subtask 7: 0 / 12
• Subtask 8: 0 / 12
• Subtask 9: 0 / 12
• Subtask 10: 0 / 12