답안 #850511

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
850511 2023-09-16T18:42:46 Z hngwlog 가로등 (APIO19_street_lamps) C++17
80 / 100
5000 ms 36664 KB
#include <bits/stdc++.h>
using namespace std;

#define fi first
#define se second
#define _size(x) (int)x.size()
#define BIT(i, x) ((x >> i) & 1)
#define MASK(n) ((1 << n) - 1)
#define REP(i, n) for (int i = 0, _n = (n); i < _n; i++)
#define FOR(i, a, b) for (int i = a, _b = (b); i <= _b; i++)
#define FORD(i, a, b) for (int i = a, _b = (b); i >= _b; i--)
#define FORB1(i, mask) for (int i = mask; i > 0; i ^= i & - i)
#define FORB0(i, n, mask) for (int i = ((1 << n) - 1) ^ mask; i > 0; i ^= i & - i)
#define FORALL(i, a) for (auto i: a)
#define fastio ios_base::sync_with_stdio(0); cin.tie(0);

struct queryNode {

    string type;
    int id, a, b;
};

int n, q;
string s;
vector<int> a;
vector<queryNode> qu;

namespace subtask1 {

    void main() {

        vector<vector<int>> c(q + 1, vector<int>(n + 1));
        FOR(i, 1, n) c[0][i] = a[i] + c[0][i - 1];
        REP(i, q) {
            FOR(j, 1, n) c[i + 1][j] = c[i][j] - c[i][j - 1];
            if (i && qu[i - 1].type == "toggle") c[i + 1][qu[i - 1].id] = 1 - c[i + 1][qu[i - 1].id];
            FOR(j, 1, n) c[i + 1][j] += c[i + 1][j - 1];
            if (qu[i].type == "query") {
                int a = qu[i].a, b = qu[i].b - 1;
                int ans = 0;
                FOR(j, 1, i + 1) ans += (c[j][b] - c[j][a - 1] == b - a + 1);
                cout << ans << '\n';
            }
        }
    }
}

bool checkSub2() {

    REP(i, q) {
        if (qu[i].type == "toggle") continue;
        if (qu[i].b - qu[i].a != 1) return false;
    }
    return true;
}

namespace subtask2 {

    void main() {

        vector<int> ans(n + 1), in(n + 1, - 1);
        FOR(i, 1, n) if (a[i] == 1) in[i] = 1;
        REP(i, q) {
            if (i && qu[i - 1].type == "toggle") {
                int id  = qu[i - 1].id;
                if (!a[id]) {
                    a[id] = 1 - a[id];
                    in[id] = i + 1;
                }
                else {
                    a[id] = 0;
                    ans[id] += (i + 1) - in[id];
                    in[id] = - 1;
                }
            }
            if (qu[i].type == "query") cout << ans[qu[i].a] + (in[qu[i].a] != - 1 ? i + 1 - in[qu[i].a] + 1 : 0) << '\n';
        }
    }
}

bool checkSub3() {

    vector<int> b(n + 1);
    FOR(i, 1, n) b[i] = a[i];
    REP(i, q) {
        if (qu[i].type == "toggle") {
            int id = qu[i].id;
            if (b[id]) return false;
            b[id] = 1 - b[id];
        }
    }
    return true;
}

namespace subtask3 {

    const int inf = 1e9;

    struct segTree {

        vector<int> ST;

        void init(int n) {

            ST.resize(4 * n + 4);
        }

        void update(int id, int l, int r, int pos, int val) {

            if (pos < l || r < pos) return ;
            if (l == r) {
                ST[id] = val;
                return ;
            }
            int mid = (l + r) / 2;
            update(id * 2, l, mid, pos, val);
            update(id * 2 + 1, mid + 1, r, pos, val);
            ST[id] = max(ST[id * 2], ST[id * 2 + 1]);
        }

        int get(int id, int l, int r, int u, int v) {

            if (r < u || v < l) return 0;
            if (u <= l && r <= v) return ST[id];
            int mid = (l + r) / 2;
            return max(get(id * 2, l, mid, u, v), get(id * 2 + 1, mid + 1, r, u, v));
        }
    } st;

    void main() {

        st.init(n);
        FOR(i, 1, n) {
            if (a[i]) st.update(1, 1, n, i, 1);
            else st.update(1, 1, n, i, inf);
        }
        REP(i, q) {
            if (qu[i].type == "toggle") {
                int id = qu[i].id;
                st.update(1, 1, n, id, i + 2);
            }
            if (qu[i].type == "query") {
                int a = qu[i].a, b = qu[i].b - 1;
                int value = st.get(1, 1, n, a, b);
                if (value == inf) cout << 0 << '\n';
                else cout << i + 1 - value + 1 << '\n';
            }
        }
    }
}

bool checkSub4() {

    int cnt = 0;
    REP(i, q) {
        if (qu[i].type == "toggle" && cnt) return false;
        cnt += (qu[i].type == "query");
    }
    return true;
}

namespace subtask4 {

    struct fenTree {

        int n;
        vector<int> bit;

        void init(int sz) {

            bit.resize(sz + 1);
            n = sz;
        }

        void update(int pos, int val) {

            while (pos <= n) {
                bit[pos] += val;
                pos += pos & - pos;
            }
        }

        int get(int pos) {

            int res = 0;
            while (pos) {
                res += bit[pos];
                pos -= pos & - pos;
            }
            return res;
        }
    } ft;

    void main() {

        map<pair<int, int>, int> in;
        set<int> st;
        int pos = - 1;
        FOR(i, 1, n) {
            if (!a[i]) {
                if (pos != - 1) {
                    in[{pos, i - 1}] = 1;
                    pos = - 1;
                }
                st.insert(i);
            }
            if (pos == - 1 && a[i]) pos = i;
        }
        if (pos) in[{pos, n}] = 1;
        vector<pair<pair<int, int>, int>> g;
        FOR(i, 1, q - 1) {
            if (qu[i - 1].type == "toggle") {
                int id = qu[i - 1].id;
                auto it = st.lower_bound(id);
                int l = - 1, r = - 1;
                if (it == st.end()) r = n;
                else r = *it - 1;
                if (it == st.begin()) l = 1;
                else {
                    it--;
                    l = *it + 1;
                }
                if (a[id]) {
                    g.push_back({{l, r}, i - in[{l, r}] + 1});
                    in.erase({l, r});
                    if (l <= id - 1) in[{l, id - 1}] = i + 1;
                    if (id + 1 <= r) in[{id + 1, r}] = i + 1;
                    st.insert(id);
                }
                else {
                    it = st.upper_bound(id);
                    if (it == st.end()) r = n;
                    else r = *it - 1;
                    if (l <= id - 1) {
                        g.push_back({{l, id - 1}, i - in[{l, id - 1}] + 1});
                        in.erase({l, id - 1});
                    }
                    if (id + 1 <= r) {
                        g.push_back({{id + 1, r}, i - in[{id + 1, r}] + 1});
                        in.erase({id + 1, r});
                    }
                    in[{l, r}] = i + 1;
                    st.erase(id);
                }
                a[id] = 1 - a[id];
            }
            if (qu[i].type == "query") break;
        }
        sort(g.begin(), g.end(), [&] (const pair<pair<int, int>, int> a, pair<pair<int, int>, int> b) { return a.fi.fi < b.fi.fi; });
        vector<int> ask;
        REP(i, q) {
            if (qu[i].type == "toggle") continue;
            ask.push_back(i);
        }
        sort(ask.begin(), ask.end(), [&] (const int i, int j) { return qu[i].a < qu[j].a; });
        vector<int> ans(q);
        ft.init(n);
        int i = 0;
        FORALL(id, ask) {
            while (i < _size(g) && g[i].fi.fi <= qu[id].a) {
                ft.update(g[i].fi.se, g[i].se);
                i++;
            }
            ans[id] += ft.get(n) - ft.get(qu[id].b - 2);
        }
        vector<int> vtx;
        FOR(i, 1, n) if (!a[i]) vtx.push_back(i);
        int cnt = 0;
        REP(i, q) {
            if (qu[i].type == "toggle") continue;
            if (!_size(vtx)) cout << ans[i] + i + 1 - in[{1, n}] + 1 << '\n';
            else {
                pos = lower_bound(vtx.begin(), vtx.end(), qu[i].a) - vtx.begin();
                if (pos == _size(vtx) || vtx[pos] > qu[i].b - 1) {
                    int l = (pos > 0 ? vtx[pos - 1] + 1 : 1);
                    int r = (pos == _size(vtx) ? n : vtx[pos] - 1);
                    cout << ans[i] + i + 1 - in[{l, r}] + 1 << '\n';
                }
                else cout << ans[i] << '\n';
            }
        }
    }
}

namespace subtask5 {

    struct fenTree {

        vector<int> bit;

        void reset(int sz) {

            n = sz;
            bit.clear();
            bit.resize(n + 1);
        }

        void update(int pos, int val) {

            while (pos <= n) {
                bit[pos] += val;
                pos += pos & - pos;
            }
        }

        int get(int pos) {

            int res = 0;
            while (pos) {
                res += bit[pos];
                pos -= pos & - pos;
            }
            return res;
        }
    } ft;

    void main() {

        /***
            chia thanh sqrt(q) block
            duyet trong block
                neu la "toggle" thi update nhu bthg
                con khong duyet lai cac doan moi them
        */

        set<int> st;
        vector<pair<int, int>> cost;
        int pos = - 1;
        FOR(i, 1, n) {
            if (!a[i]) {
                if (pos != - 1) {
                    cost.push_back({pos, i - 1});
                    pos = - 1;
                }
                st.insert(i);
            }
            if (pos == - 1 && a[i]) pos = i;
        }
        if (pos != - 1) cost.push_back({pos, n});

        /***

        */
        vector<int> b(n + 1);
        FOR(i, 1, n) b[i] = a[i];
        REP(i, q) {
            if (qu[i].type == "query") continue;
            int id = qu[i].id;
            auto it = st.lower_bound(id);
            int l = - 1, r = - 1;
            if (it == st.end()) r = n;
            else r = *it - 1;
            if (it == st.begin()) l = 1;
            else {
                it--;
                l = *it + 1;
            }
            if (b[id]) {
                if (l <= id - 1) cost.push_back({l, id - 1});
                if (id + 1 <= r) cost.push_back({id + 1, r});
                st.insert(id);
            }
            else {
                it = st.upper_bound(id);
                if (it == st.end()) r = n;
                else r = *it - 1;
                if (l <= id - 1) cost.push_back({l, id - 1});
                if (id + 1 <= r) cost.push_back({id + 1, r});
                cost.push_back({l, r});
                st.erase(id);
            }
            b[id] = 1 - b[id];
        }
        sort(cost.begin(), cost.end());
        cost.resize(unique(cost.begin(), cost.end()) - cost.begin());
        vector<vector<int>> adj(q);
        while (_size(st)) st.erase(st.begin());
        FOR(i, 1, n) {
            b[i] = a[i];
            if (!b[i]) st.insert(i);
        }
        REP(i, q) {
            if (qu[i].type == "query") continue;
            int id = qu[i].id;
            auto it = st.lower_bound(id);
            int l = - 1, r = - 1;
            if (it == st.end()) r = n;
            else r = *it - 1;
            if (it == st.begin()) l = 1;
            else {
                it--;
                l = *it + 1;
            }
            if (b[id]) {
                adj[i].push_back(lower_bound(cost.begin(), cost.end(), make_pair(l, r)) - cost.begin());
                if (l <= id - 1) adj[i].push_back(lower_bound(cost.begin(), cost.end(), make_pair(l, id - 1)) - cost.begin());
                if (id + 1 <= r) adj[i].push_back(lower_bound(cost.begin(), cost.end(), make_pair(id + 1, r)) - cost.begin());
                st.insert(id);
            }
            else {
                it = st.upper_bound(id);
                if (it == st.end()) r = n;
                else r = *it - 1;
                if (l <= id - 1) adj[i].push_back(lower_bound(cost.begin(), cost.end(), make_pair(l, id - 1)) - cost.begin());
                if (id + 1 <= r) adj[i].push_back(lower_bound(cost.begin(), cost.end(), make_pair(id + 1, r)) - cost.begin());
                adj[i].push_back(lower_bound(cost.begin(), cost.end(), make_pair(l, r)) - cost.begin());
                st.erase(id);
            }
            b[id] = 1 - b[id];
        }
        while (_size(st)) st.erase(st.begin());
        vector<int> in(_size(cost));
        pos = - 1;
        FOR(i, 1, n) {
            if (!a[i]) {
                if (pos != - 1) {
                    in[lower_bound(cost.begin(), cost.end(), make_pair(pos, i - 1)) - cost.begin()] = 1;
                    pos = - 1;
                }
                st.insert(i);
            }
            if (pos == - 1 && a[i]) pos = i;
        }
        if (pos != - 1) in[lower_bound(cost.begin(), cost.end(), make_pair(pos, n)) - cost.begin()] = 1;

        /***

        */

        int blockSz = 2000;
        int cnt = q / blockSz + (q % blockSz ? 1 : 0);
        vector<int> ans(q);
        vector<pair<int, int>> g;
        REP(t, cnt) {
            int l = t * blockSz, r = min(q - 1, (t + 1) * blockSz - 1);
            vector<int> ask, change;
            FOR(i, l, r) if (qu[i].type == "query") ask.push_back(i); else change.push_back(i);
            sort(ask.begin(), ask.end(), [&] (const int i, int j) { return qu[i].a < qu[j].a; });
            sort(g.begin(), g.end(), [&] (const pair<int, int> a, pair<int, int> b) { return cost[a.fi].fi < cost[b.fi].fi; });
            ft.reset(n);
            int j = 0;
            FORALL(id, ask) {
                while (j < _size(g) && cost[g[j].fi].fi <= qu[id].a) {
                    ft.update(n - cost[g[j].fi].se + 1, g[j].se);
                    j++;
                }
                vector<int> sav_vtx;
                vector<pair<int, int>> sav_in, _g;
                FORALL(i, change) {
                    if (i > id) break;
                    int _id = qu[i].id;
                    auto it = st.lower_bound(_id);
                    int l = - 1, r = - 1;
                    if (it == st.end()) r = n;
                    else r = *it - 1;
                    if (it == st.begin()) l = 1;
                    else {
                        it--;
                        l = *it + 1;
                    }
                    int cnt = 0;
                    if (a[_id]) {
                        _g.push_back({adj[i][cnt], i + 1 - in[adj[i][cnt]] + 1});
                        sav_in.push_back({adj[i][cnt], - in[adj[i][cnt]]});
                        if (l <= _id - 1) {
                            cnt++;
                            in[adj[i][cnt]] = i + 2;
                            sav_in.push_back({adj[i][cnt], i + 1});
                        }
                        if (_id + 1 <= r) {
                            cnt++;
                            in[adj[i][cnt]] = i + 2;
                            sav_in.push_back({adj[i][cnt], i + 1});
                        }
                        st.insert(_id);
                        sav_vtx.push_back(_id);
                    }
                    else {
                        it = st.upper_bound(_id);
                        if (it == st.end()) r = n;
                        else r = *it - 1;
                        if (l <= _id - 1) {
                            _g.push_back({adj[i][cnt], i + 1 - in[adj[i][cnt]] + 1});
                            sav_in.push_back({adj[i][cnt], - in[adj[i][cnt]]});
                            cnt++;
                        }
                        if (_id + 1 <= r) {
                            _g.push_back({adj[i][cnt], i + 1 - in[adj[i][cnt]] + 1});
                            sav_in.push_back({adj[i][cnt], - in[adj[i][cnt]]});
                            cnt++;
                        }
                        in[adj[i][cnt]] = i + 2;
                        sav_in.push_back({adj[i][cnt], i + 2});
                        st.erase(_id);
                        sav_vtx.push_back(- _id);
                    }
                    a[_id] = 1 - a[_id];
                }
                int res = ft.get(n - (qu[id].b - 1) + 1);
                FORALL(e, _g) if (cost[e.fi].fi <= qu[id].a && cost[e.fi].se >= qu[id].b - 1) res += e.se;
                auto it = st.lower_bound(qu[id].a);
                if (it == st.end() || *it > qu[id].b - 1) {
                    int pos = - 1;
                    if (it == st.begin()) pos = 1;
                    else {
                        it--;
                        pos = *it + 1;
                        it++;
                    }
                    int rpos = - 1;
                    if (it == st.end()) rpos = n;
                    else rpos = *it - 1;
                    res += id + 1 - in[lower_bound(cost.begin(), cost.end(), make_pair(pos, rpos)) - cost.begin()] + 1;
                }
                ans[id] = res;
                FORD(i, _size(sav_in) - 1, 0) in[sav_in[i].fi] = abs(sav_in[i].se);
                FORD(i, _size(sav_vtx) - 1, 0) {
                    int pos = sav_vtx[i];
                    a[abs(pos)] = 1 - a[abs(pos)];
                    if (pos < 0) st.insert(- pos);
                    else st.erase(pos);
                }
            }
            FOR(i, l, r) {
                if (qu[i].type == "query") continue;
                int id = qu[i].id;
                auto it = st.lower_bound(id);
                int l = - 1, r = - 1;
                if (it == st.end()) r = n;
                else r = *it - 1;
                if (it == st.begin()) l = 1;
                else {
                    it--;
                    l = *it + 1;
                }
                int cnt = 0;
                if (a[id]) {
                    g.push_back({adj[i][cnt], i + 1 - in[adj[i][cnt]] + 1});
                    if (l <= id - 1) in[adj[i][++cnt]] = i + 2;
                    if (id + 1 <= r) in[adj[i][++cnt]] = i + 2;
                    st.insert(id);
                }
                else {
                    it = st.upper_bound(id);
                    if (it == st.end()) r = n;
                    else r = *it - 1;
                    if (l <= id - 1) g.push_back({adj[i][cnt], i + 1 - in[adj[i][cnt++]] + 1});
                    if (id + 1 <= r) g.push_back({adj[i][cnt], i + 1 - in[adj[i][cnt++]] + 1});
                    in[adj[i][cnt]] = i + 2;
                    st.erase(id);
                }
                a[id] = 1 - a[id];
            }
        }
        REP(i, q) if (qu[i].type == "query") cout << ans[i] << '\n';
    }
}

int main() {
    fastio;

    cin >> n >> q;
    cin >> s;
    a.resize(n + 1);
    FOR(i, 1, n) a[i] = s[i - 1] - '0';
    qu.resize(q);
    REP(i, q) {
        cin >> qu[i].type;
        if (qu[i].type == "toggle") cin >> qu[i].id;
        else cin >> qu[i].a >> qu[i].b;
    }
    if (n <= 100 && q <= 100) subtask1::main();
    else if (checkSub2()) subtask2::main();
    else if (checkSub3()) subtask3::main();
    else if (checkSub4()) subtask4::main();
    else subtask5::main();
    return 0;
}

Compilation message

street_lamps.cpp: In function 'void subtask4::main()':
street_lamps.cpp:268:13: warning: unused variable 'cnt' [-Wunused-variable]
  268 |         int cnt = 0;
      |             ^~~
street_lamps.cpp: In function 'void subtask5::main()':
street_lamps.cpp:547:85: warning: operation on 'cnt' may be undefined [-Wsequence-point]
  547 |                     if (l <= id - 1) g.push_back({adj[i][cnt], i + 1 - in[adj[i][cnt++]] + 1});
      |                                                                                  ~~~^~
street_lamps.cpp:548:85: warning: operation on 'cnt' may be undefined [-Wsequence-point]
  548 |                     if (id + 1 <= r) g.push_back({adj[i][cnt], i + 1 - in[adj[i][cnt++]] + 1});
      |                                                                                  ~~~^~
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 1 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 344 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 65 ms 15304 KB Output is correct
2 Correct 69 ms 15560 KB Output is correct
3 Correct 82 ms 15328 KB Output is correct
4 Correct 81 ms 19172 KB Output is correct
5 Correct 78 ms 19456 KB Output is correct
6 Correct 70 ms 18916 KB Output is correct
7 Correct 80 ms 18940 KB Output is correct
8 Correct 84 ms 20456 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 1 ms 348 KB Output is correct
5 Correct 132 ms 20588 KB Output is correct
6 Correct 154 ms 20820 KB Output is correct
7 Correct 180 ms 20944 KB Output is correct
8 Correct 207 ms 22600 KB Output is correct
9 Correct 74 ms 12628 KB Output is correct
10 Correct 100 ms 13652 KB Output is correct
11 Correct 81 ms 13648 KB Output is correct
12 Correct 204 ms 21372 KB Output is correct
13 Correct 227 ms 22680 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 348 KB Output is correct
2 Correct 1 ms 348 KB Output is correct
3 Correct 1 ms 348 KB Output is correct
4 Correct 1 ms 348 KB Output is correct
5 Correct 208 ms 33396 KB Output is correct
6 Correct 284 ms 33812 KB Output is correct
7 Correct 328 ms 34416 KB Output is correct
8 Correct 445 ms 36664 KB Output is correct
9 Correct 162 ms 19264 KB Output is correct
10 Correct 155 ms 22460 KB Output is correct
11 Correct 163 ms 19496 KB Output is correct
12 Correct 150 ms 22472 KB Output is correct
13 Correct 165 ms 19276 KB Output is correct
14 Correct 150 ms 22728 KB Output is correct
15 Correct 208 ms 21368 KB Output is correct
16 Correct 205 ms 22604 KB Output is correct
# 결과 실행 시간 메모리 Grader output
1 Correct 1 ms 344 KB Output is correct
2 Correct 0 ms 348 KB Output is correct
3 Correct 0 ms 344 KB Output is correct
4 Correct 1 ms 348 KB Output is correct
5 Correct 0 ms 348 KB Output is correct
6 Correct 0 ms 344 KB Output is correct
7 Correct 0 ms 348 KB Output is correct
8 Correct 65 ms 15304 KB Output is correct
9 Correct 69 ms 15560 KB Output is correct
10 Correct 82 ms 15328 KB Output is correct
11 Correct 81 ms 19172 KB Output is correct
12 Correct 78 ms 19456 KB Output is correct
13 Correct 70 ms 18916 KB Output is correct
14 Correct 80 ms 18940 KB Output is correct
15 Correct 84 ms 20456 KB Output is correct
16 Correct 1 ms 348 KB Output is correct
17 Correct 1 ms 348 KB Output is correct
18 Correct 1 ms 348 KB Output is correct
19 Correct 1 ms 348 KB Output is correct
20 Correct 132 ms 20588 KB Output is correct
21 Correct 154 ms 20820 KB Output is correct
22 Correct 180 ms 20944 KB Output is correct
23 Correct 207 ms 22600 KB Output is correct
24 Correct 74 ms 12628 KB Output is correct
25 Correct 100 ms 13652 KB Output is correct
26 Correct 81 ms 13648 KB Output is correct
27 Correct 204 ms 21372 KB Output is correct
28 Correct 227 ms 22680 KB Output is correct
29 Correct 1 ms 348 KB Output is correct
30 Correct 1 ms 348 KB Output is correct
31 Correct 1 ms 348 KB Output is correct
32 Correct 1 ms 348 KB Output is correct
33 Correct 208 ms 33396 KB Output is correct
34 Correct 284 ms 33812 KB Output is correct
35 Correct 328 ms 34416 KB Output is correct
36 Correct 445 ms 36664 KB Output is correct
37 Correct 162 ms 19264 KB Output is correct
38 Correct 155 ms 22460 KB Output is correct
39 Correct 163 ms 19496 KB Output is correct
40 Correct 150 ms 22472 KB Output is correct
41 Correct 165 ms 19276 KB Output is correct
42 Correct 150 ms 22728 KB Output is correct
43 Correct 208 ms 21368 KB Output is correct
44 Correct 205 ms 22604 KB Output is correct
45 Execution timed out 5021 ms 21808 KB Time limit exceeded
46 Halted 0 ms 0 KB -