洛谷P1972 [SDOI2009]HH的项链(树状数组)

时间:2023-11-19 08:44:43

题目链接:

https://www.luogu.org/problemnew/show/P1972

题目描述:

HH 有一串由各种漂亮的贝壳组成的项链。HH 相信不同的贝壳会带来好运,所以每次散步完后,他都会随意取出一段贝壳,思考它们所表达的含义。HH 不断地收集新的贝壳,因此,他的项链变得越来越长。有一天,他突然提出了一个问题:某一段贝壳中,包含了多少种不同的贝壳?这个问题很难回答……因为项链实在是太长了。于是,他只好求助睿智的你,来解决这个问题。

输入输出格式

输入格式:

第一行:一个整数N,表示项链的长度。

第二行:N 个整数,表示依次表示项链中贝壳的编号(编号为0 到1000000 之间的整数)。

第三行:一个整数M,表示HH 询问的个数。

接下来M 行:每行两个整数,L 和R(1 ≤ L ≤ R ≤ N),表示询问的区间。

输出格式:

M 行,每行一个整数,依次表示询问对应的答案。

输入输出样例

输入样例#1:
6
1 2 3 4 3 5
3
1 2
3 5
2 6
输出样例#1:
2
2
4

说明

数据范围:

对于100%的数据,N <= 500000,M <= 500000。

解题思路:

离线处理,边更新边查询。以右端点升序更新所有查询的区间。

 1 #include <iostream>
2 #include<cstdio>
3 #include<algorithm>
4 #define lowbit(x) (x&(-x))
5 using namespace std;//区间查询
6 int C[500010];
7 int ans[500010],num[500010];
8 int booll[1000010];//位置
9 struct Quiry{
10 int l,r;
11 int pos;
12 bool operator < (const Quiry q) const{
13 return r<q.r;//只需要保证右边升序即可
14 }
15 }q[500010];
16 void add(int x,int v){
17 for(int i=x;i<=500010;i+=lowbit(i)){
18 C[i]+=v;
19 }
20 }
21 int getSum(int x){
22 int sum=0;
23 for(int i=x;i>0;i-=lowbit(i)){
24 sum+=C[i];
25 }
26 return sum;
27 }
28 int main(int argc, char** argv) {
29 //离线做
30 int n;
31 scanf("%d",&n);
32 for(int i=1;i<=n;i++){
33 scanf("%d",&num[i]);
34 }
35 int m;
36 scanf("%d",&m);
37 for(int i=1;i<=m;i++){
38 scanf("%d %d",&q[i].l,&q[i].r);
39 q[i].pos=i;
40 }
41 sort(q+1,q+m+1);
42 //边更新树状数组,边查询
43 int next=1;
44 for(int i=1;i<=m;i++){
45 for(int j=next;j<=q[i].r;j++){
46 if(booll[num[j]]){
47 add(booll[num[j]],-1);
48 }
49 add(j,1);//位置
50 booll[num[j]]=j;
51 }
52 next=q[i].r+1;
53
54 ans[q[i].pos]=getSum(q[i].r)-getSum(q[i].l-1);
55 }
56 for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
57
58 return 0;
59 }

参考自:https://www.luogu.org/blog/little-black/solution-p1972