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;
}
本题也有在线做法,需要用到可持久化线段树。
浙公网安备 33010602011771号