chenfy27的刷题记录

导航

abc174F Range Set Query

给定数组A[N],有Q个询问,每个询问给出l[i]和r[i],问区间[l[i],r[i]]内有多少个不同的数?
1<=N,Q<=5E5; 1<=A[i]<=N; 1<=l[i]<=r[i]<=N

分析:对询问按右端点从小到大排序,然后从左到右依次处理每个A[i],将下标i的位置置为1,如果前面出现过A[i],则把上一次出现的位置置为0,然后处理右端点为i的询问,其实就是个前缀和问题,可以用BIT来维护,这里用的平衡树。

#include <bits/stdc++.h>
using i64 = long long;

// Treap模板...

struct Node {
    int l, r, x, ans;
};

void solve() {
    int N, Q;
    std::cin >> N >> Q;
    std::vector<int> c(N + 1);
    for (int i = 1; i <= N; i++) {
        std::cin >> c[i];
    }
    std::vector<Node> qry(Q + 1);
    for (int i = 1; i <= Q; i++) {
        std::cin >> qry[i].l >> qry[i].r;
        qry[i].x = i;
    }
    std::sort(qry.begin() + 1, qry.end(), [](Node &u, Node &v){
        return u.r < v.r;
    });

    std::vector<int> lst(N + 1);
    Treap<int> tr;
    for (int i = 1, j = 1; i <= N; i++) {
        if (lst[c[i]]) {
            tr.erase(lst[c[i]]);
        }
        tr.insert(i);
        while (j <= Q && qry[j].r <= i) {
            qry[j].ans = tr.lecnt(qry[j].r) - tr.ltcnt(qry[j].l);
            j += 1;
        }
        lst[c[i]] = i;
    }

    std::sort(qry.begin() + 1, qry.end(), [](Node &u, Node &v){
        return u.x < v.x;
    });
    for (int i = 1; i <= Q; i++) {
        std::cout << qry[i].ans << "\n";
    }
}

int main() {
    std::cin.tie(0)->sync_with_stdio(0);
    int t = 1;
    while (t--) solve();
    return 0;
}

本题也有在线做法,需要用到可持久化线段树。

posted on 2024-10-09 19:56  chenfy27  阅读(17)  评论(0)    收藏  举报