QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#419786#6533. Traveling in CellsXiaoretaWTL 188ms3636kbC++203.7kb2024-05-24 11:15:442024-05-24 11:15:45

Judging History

你现在查看的是最新测评结果

  • [2024-05-24 11:15:45]
  • 评测
  • 测评结果:TL
  • 用时:188ms
  • 内存:3636kb
  • [2024-05-24 11:15:44]
  • 提交

answer

#include <bits/stdc++.h>
using namespace std;
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define sz(a) ((int)a.size())
#define all(a) a.begin(), a.end()
#define rep(i, l, r) for (int i = l; i < r; ++i)
#define per(i, r, l) for (int i = r-1; i >= l; --i)
typedef long long ll;
typedef pair<int, int> PI;
template<typename T> bool setmax(T &a, T b) { return (a < b ? a = b, 1 : 0); }
template<typename T> bool setmin(T &a, T b) { return (a > b ? a = b, 1 : 0); }

// // 比 unordered_map 更快的哈希表
// #include <ext/pb_ds/assoc_container.hpp>
// using namespace __gnu_pbds;
// const int RANDOM = chrono::high_resolution_clock::now().time_since_epoch().count();
// struct chash {
//     int operator()(int x) const { return x ^ RANDOM; }
// };
// typedef gp_hash_table<int, int, chash> hash_t;

template<typename T> struct BIT {
    int n;
    vector<T> tr;
    BIT(int _n) {
        n = _n;
        tr.resize(n+1);
    }
    T prefix(int i) {
        T res = 0;
        for (; i; i -= i&(-i)) res += tr[i];
        return res;
    }
    T query(int l, int r) { return prefix(r) - prefix(l-1); }
    void modify(int i, T val) { for (; i <= n; i += i&(-i)) tr[i] += val; }
};

// const int N = 300010;
// hash_t bitc[N];

// void modify(int c, int p, int val) {
// }

void solve() {
    int n, q; cin >> n >> q;
    vector<int> c(n + 1), v(n + 1);
    vector<BIT<int>> cbit(n + 1, BIT<int>(n));
    BIT<ll> vbit(n);
    for (int i = 1; i <= n; i++) {
        cin >> c[i];
        cbit[c[i]].modify(i, 1);
    }
    for (int i = 1; i <= n; i++) {
        cin >> v[i];
        vbit.modify(i, v[i]);
    }
    while (q--) {
        int t; cin >> t;
        if (t == 1 || t == 2) {
            int p, x; cin >> p >> x;
            if (t == 1) {
                cbit[c[p]].modify(p, -1);
                c[p] = x;
                cbit[c[p]].modify(p, 1);
            } else {
                vbit.modify(p, -v[p]);
                v[p] = x;
                vbit.modify(p, v[p]);
            }
        } else {
            int x, k; cin >> x >> k;
            vector<int> st(k);
            for (int &i : st) cin >> i;
            auto gao1 = [&]() {
                auto ck1 = [&](int m) {
                    int tot = 0;
                    for (int color : st) {
                        // len = m
                        // [x - m + 1, x];
                        tot += cbit[color].query(x - m + 1, x);
                    }
                    assert(tot <= m);
                    return tot == m;
                };
                int l = 1, r = x;
                while (l < r) {
                    int mid = (l + r + 1) >> 1;
                    if (ck1(mid)) l = mid;
                    else r = mid - 1;
                }
                return l;
            };
            auto gao2 = [&]() {
                auto ck2 = [&](int m) {
                    int tot = 0;
                    for (int color : st) {
                        tot += cbit[color].query(x, x + m - 1);
                    }
                    assert(tot <= m);
                    return tot == m;
                };
                int l = 1, r = n - x + 1;
                while (l < r) {
                    int mid = (l + r + 1) >> 1;
                    if (ck2(mid)) l = mid;
                    else r = mid - 1;
                }
                return l;
            };
            int L = x - gao1() + 1, R = x + gao2() - 1;
            cout << vbit.query(L, R) << '\n';
        }
    }
}
int main() {
    ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
    int tt; cin >> tt;
    while (tt--) solve();

    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 0ms
memory: 3612kb

input:

2
5 10
1 2 3 1 2
1 10 100 1000 10000
3 3 1 3
3 3 2 2 3
2 5 20000
2 3 200
3 3 2 1 3
3 3 3 1 2 3
1 3 4
2 1 100000
1 2 2
3 1 2 1 2
4 1
1 2 3 4
1000000 1000000 1000000 1000000
3 4 4 1 2 3 4

output:

100
110
1200
21211
100010
4000000

result:

ok 6 numbers

Test #2:

score: 0
Accepted
time: 128ms
memory: 3544kb

input:

20000
15 15
1 1 3 3 2 3 3 3 3 3 2 3 2 3 3
634593973 158136379 707704004 998369647 831633445 263092797 937841412 451774682 552617416 483763379 50360475 794662797 74247465 537217773 901809831
3 6 4 1 3 5 10
3 5 7 1 2 3 4 5 9 10
3 4 3 3 8 9
2 13 608033129
3 15 2 3 5
1 9 3
3 8 4 1 3 7 10
2 6 727472865
3...

output:

2689089686
8377825475
1706073651
1439027604
2689089686
792730352
8904867081
8904867081
8270273108
831633445
692051585
2782432626
697783016
883944422
184057757
287523250
184057757
696388752
184057757
1675459344
2667693025
2614711258
4006992373
1767091974
5348541057
5348541057
390631780
2290216252
942...

result:

ok 200062 numbers

Test #3:

score: 0
Accepted
time: 188ms
memory: 3636kb

input:

2000
150 150
8 3 8 8 8 6 8 4 2 7 6 8 8 5 8 7 7 8 5 6 8 8 6 8 8 8 8 7 8 6 6 8 8 8 6 2 3 4 8 8 7 8 5 8 2 6 8 7 8 8 6 8 6 8 3 8 8 8 8 4 7 8 7 3 7 6 7 5 5 8 6 8 8 6 3 8 6 7 6 8 8 7 4 8 6 7 8 7 7 7 7 8 8 8 8 2 5 2 8 8 6 7 6 3 8 8 7 8 8 8 6 6 8 6 6 7 5 8 8 8 7 8 7 7 6 8 8 8 8 8 8 6 5 7 5 5 8 7 8 7 7 7 6 5...

output:

4449391171
3290849667
852793841
5178673994
995994209
11431868919
4327723427
5071541023
3032743466
962345334
2997656427
4534278452
3851900075
3611231417
5071541023
1477584218
1299005818
1299005818
2145605244
854143763
886347565
2081234124
2333808475
2455955801
4179722063
2328504333
1473735464
4107685...

result:

ok 199987 numbers

Test #4:

score: -100
Time Limit Exceeded

input:

10
30000 30000
3 4 2 4 4 4 4 3 4 3 4 3 4 3 4 4 2 4 4 4 4 4 3 3 3 4 3 4 3 4 3 3 4 2 4 3 3 3 3 4 3 4 4 4 4 2 3 3 4 2 3 4 4 4 4 1 4 4 4 4 4 4 4 4 3 3 3 4 4 4 4 4 2 3 4 4 4 4 3 4 4 3 3 3 4 4 3 4 4 2 3 4 4 4 4 3 2 4 3 4 3 2 4 4 3 4 2 2 4 4 4 4 2 4 3 2 4 4 3 4 4 4 2 4 4 3 2 3 2 3 3 3 4 2 4 3 4 1 4 3 4 4 4...

output:

6959437173
934970676
72461245502
8365928740
8384151048
984567228
12482909122
1904927816
15134139942
3759040688
92670874909
332468911
5936663371
3562978848
1300592004
10314009201
5581540905
131246926443
15087184135864
4077066271
1124704817
1520626740
4388174158
744377942
2770411457
6231852240
1508724...

result: