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;
}

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