QOJ.ac

QOJ

IDProblemSubmitterResultTimeMemoryLanguageFile sizeSubmit timeJudge time
#384344#6533. Traveling in Cellsmai123WA 229ms5680kbC++144.8kb2024-04-09 22:12:032024-04-09 22:12:03

Judging History

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

  • [2024-04-09 22:12:03]
  • 评测
  • 测评结果:WA
  • 用时:229ms
  • 内存:5680kb
  • [2024-04-09 22:12:03]
  • 提交

answer

#include<bits/stdc++.h>

#define int long long
#define ios ios::sync_with_stdio(false),cin.tie(nullptr),cout.tie(nullptr)
using namespace std;
typedef pair<int, int> PII;
const int Maxn = 1e5 + 10;
int c[Maxn], v[Maxn];
unordered_map<int, unordered_set<int>> pos;//记录每个颜色的位置
// 树状数组维护区间和
int tr[Maxn];

int lowbit(int x) {
    return x & -x;
}

void add(int x, int d) {
    for (; x < Maxn; x += lowbit(x))tr[x] += d;
}

int sum(int x) {
    int res = 0;
    for (; x > 0; x -= lowbit(x))res += tr[x];
    return res;
}


struct Node {
    int l, r, val;
} T[Maxn << 2];

void pushUp(int node) {
    T[node].val = T[node << 1].val + T[node << 1 | 1].val;
}

void build(int node, int l, int r) {
    T[node] = {l, r, 0};
    if (l == r)return;
    int mid = (l + r) >> 1;
    build(node << 1, l, mid);
    build(node << 1 | 1, mid + 1, r);
    pushUp(node);
}

void update(int node, int l, int r, int d) {
    if (l <= T[node].l && T[node].r <= r) {
        T[node].val = d;
        return;
    }
    int mid = (T[node].l + T[node].r) >> 1;
    if (l <= mid)update(node << 1, l, r, d);
    if (mid < r)update(node << 1 | 1, l, r, d);
    pushUp(node);
}

// 查询[l,r]区间内,最左边的完全填充区
int query_left(int node, int l, int r) {
    int mid = (T[node].l + T[node].r) >> 1;
    // 该节点被查询区间完全包括
    if (l <= T[node].l && T[node].r <= r) {
        if (T[node].val == T[node].r - T[node].l + 1) {
            return T[node].l;
        }
        // 区间未完全填充

        if (T[node].l == T[node].r)return T[node].r + 1;
        // 先查询靠近x的区域
        int left = query_left(node << 1 | 1, l, r);
        if (left == mid + 1) left = query_left(node << 1, l, r);
        return left;
    } else if (l <= mid && mid < r) {
        // 查询区间被当前节点包括,且中点被查询区间包括
        int left = query_left(node << 1 | 1, l, r);
        if (left == mid + 1)left = query_left(node << 1, l, r);
        return left;
    } else if (l <= mid)return query_left(node << 1, l, r);//查左子树
    else if (mid < r)return query_left(node << 1 | 1, l, r);//查右子树
}

// 查询[l,r]区间内,最右边的完全填充区
int query_right(int node, int l, int r) {
    int mid = (T[node].l + T[node].r) >> 1;
    // 该节点被查询区间完全包括
    if (l <= T[node].l && T[node].r <= r) {
        if (T[node].val == T[node].r - T[node].l + 1) {
            return T[node].r;
        }
        // 该区间未完全填充
        if (T[node].l == T[node].r)return T[node].l - 1;
        int right = query_right(node << 1, l, r);
        if (right == mid + 1)right = query_right(node << 1 | 1, l, r);
        return right;
    } else if (l <= mid && mid < r) {
        // 查询区间被当前节点包括,且中点被查询区间包括
        int right = query_right(node << 1, l, r);
        if (right == mid)right = query_right(node << 1 | 1, l, r);
        return right;
    } else if (l <= mid)return query_right(node << 1, l, r);//查左子树
    else if (mid < r)return query_right(node << 1 | 1, l, r);//查右子树
}


signed main() {
    ios;
    int t;
    cin >> t;
    while (t--) {
        pos.clear();
        int n, q;
        cin >> n >> q;
        for (int i = 1; i <= n; i++)tr[i] = 0;
        for (int i = 1; i <= n; i++) {
            cin >> c[i];
            pos[c[i]].insert(i);//记录每种颜色的所有位置
        }
        // 树状数组维护的是区间和
        for (int i = 1; i <= n; i++) {
            cin >> v[i];
            add(i, v[i]);
        }
        while (q--) {
            int op, p, x, k;
            cin >> op;
            if (op == 1) {
                cin >> p >> x;
                // cp改成x
                pos[c[p]].erase(p);
                if (pos[c[p]].empty())pos.erase(c[p]);
                pos[x].insert(p);
                c[p] = x;
            } else if (op == 2) {
                cin >> p >> x;
                // vp改成x
                int diff = x - v[p];
                add(p, diff);
                v[p] = x;
            } else {
                build(1, 1, n);
                cin >> x >> k;
                for (int i = 0; i < k; i++) {
                    int a;
                    cin >> a;
                    // 遍历颜色a的所有位置,并 +1
                    for (int c_pos: pos[a]) {
                        update(1, c_pos, c_pos, 1);
                    }
                }
                int left = query_left(1, 1, x);
                int right = query_right(1, x, n);
                int ans = 0;
                if(left <=x)ans -= sum(left-1);
                if(x<=right)ans += sum(right);
                cout << ans << endl;
            }
        }
    }
    return 0;
}

Details

Tip: Click on the bar to expand more detailed information

Test #1:

score: 100
Accepted
time: 1ms
memory: 5676kb

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: -100
Wrong Answer
time: 229ms
memory: 5680kb

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
4482602664
390631780
2290216252
942...

result:

wrong answer 26th numbers differ - expected: '5348541057', found: '4482602664'