Submission #934351

#	Time	Username	Problem	Language	Result	Execution time	Memory
934351		vjudge1	Street Lamps (APIO19_street_lamps)	C++17	20 / 100	80 ms	11604 KiB

street_lamps

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

const int mxN = 3e5 + 7;

int n, q, arr[mxN], res[mxN], on[mxN];

int32_t main() {
    ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);

    cin >> n >> q;
    for (int i = 1; i <= n; ++i) {
        char c;
        cin >> c;

        if (c == '1') {
            arr[i] = 1;
        }
        on[i] = -1;
    }

    for (int i = 1; i <= n; ++i) {
        if (arr[i]) {
            on[i] = 0;
        }
    }
    
    for (int i = 1; i <= q; ++i) {
        string t;
        cin >> t;

        if (t[0] == 'q') {
            int a, b;
            cin >> a >> b;
            
            cout << res[a] + (on[a] != -1 ? i - on[a] : 0) << "\n";
        }
        else {
            int ind;
            cin >> ind;
            arr[ind] ^= 1;

            if (arr[ind]) {
                assert(on[ind] == -1);
                on[ind] = i;
            }
            else {
                res[ind] += i - on[ind];
                on[ind] = -1;
            }
        }
    }
}

• Subtask 1: 0 / 20
• Subtask 2: 20 / 20
• Subtask 3: 0 / 20
• Subtask 4: 0 / 20
• Subtask 5: 0 / 20