Submission #1276962

#	Time	Username	Problem	Language	Result	Execution time	Memory
1276962		SnowRaven52	Bob (COCI14_bob)	C++20	120 / 120	78 ms	576 KiB

bob

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

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    int n, m; cin >> n >> m;
    vector<int> pr(m, INT_MIN), cr(m), h(m, 0);
    long long ans = 0;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) cin >> cr[j];
        for (int j = 0; j < m; j++) {
            if (i > 0 && cr[j] == pr[j]) h[j] = h[j] + 1;
            else h[j] = 1;
        }
        int j = 0;
        while (j < m) {
            int k = j + 1;
            while (k < m && cr[k] == cr[j]) k++;
            long long rs = 0;
            vector<pair<int, int>> st;
            for (int x = j; x < k; x++) {
                int c = 1;
                while (!st.empty() && st.back().first >= h[x]) {
                    rs -= 1LL * st.back().first * st.back().second;
                    c += st.back().second;
                    st.pop_back();
                }
                st.emplace_back(h[x], c);
                rs += 1LL * h[x] * c;
                ans += rs;
            }
            j = k;
        }
        pr.swap(cr);
    }
    cout << ans << endl;
}

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