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

题目链接:

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

猜你喜欢

转载自www.cnblogs.com/zhuixunfighting/p/10230294.html