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

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

标签：离线 平衡树