2019CCPC湖南全国邀请赛-Chika and Friendly Pairs- 莫队+树状数组+离散化

题目链接：

http://acm.hdu.edu.cn/showproblem.php?pid=6534

思路：

　　莫队算法，树状数组求和。每个数对应的离散化之后的值需要打表存下来。

Code：

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int N=27000+10;
 4 int a[N],pos[N],disc[N],n,m,k,block;
 5 
 6 struct Query
 7 {
 8     bool operator <(const Query&t)const
 9     {
10         return pos[l]^pos[t.l]?(l<t.l):(pos[l]&1?r<t.r:t.r<r);
11     }
12     int l,r,id;
13 };
14 Query q[N];
15 
16 int c[N],tot=1;
17 inline void add(int v,int x)
18 {
19     while(x<=tot)
20         c[x]+=v,x+=x&-x;
21 }
22 int ask(int x)
23 {
24     int ans=0;
25     while(x)
26         ans+=c[x],x-=x&-x;
27     return ans;
28 }
29 
30 int res[N],up[N],down[N],v[N];
31 int main()
32 {
33     scanf("%d%d%d",&n,&m,&k);
34     for(int i=1;i<=n;++i)
35         scanf("%d",a+i);
36     for(int i=0;i<m;++i)
37     {
38         scanf("%d%d",&q[i].l,&q[i].r);
39         q[i].id=i;
40     }
41 
42     block=sqrt(n);
43     for(int i=1;i<=n;++i)
44     {
45         pos[i]=(i+block-1)/block;
46         disc[i]=a[i];
47     }
48     sort(disc+1,disc+n+1);
49     tot=unique(disc+1,disc+n+1)-disc-1;
50     sort(q,q+m);
51 
52     for(int i=1;i<=n;++i)
53     {
54         down[i]=lower_bound(disc+1,disc+tot+1,a[i]-k)-disc-1;
55         up[i]=lower_bound(disc+1,disc+tot+1,a[i]+k)-disc;
56         if(disc[up[i]]!=a[i]+k)--up[i];
57         v[i]=lower_bound(disc+1,disc+tot+1,a[i])-disc;
58     }
59 
60     int ans=0,l=1,r=1;
61     add(1,v[l]);
62     for(int i=0;i<m;++i)
63     {
64         Query &t=q[i];
65         while(r<t.r)
66         {
67             ++r;
68             ans+=ask(up[r])-ask(down[r]);
69             add(1,v[r]);
70         }
71         while(l>t.l)
72         {
73             --l;
74             ans+=ask(up[l])-ask(down[l]);
75             add(1,v[l]);
76         }
77         while(r>t.r)
78         {
79             add(-1,v[r]);
80             ans-=ask(up[r])-ask(down[r]);
81             --r;
82         }
83         while(l<t.l)
84         {
85             add(-1,v[l]);
86             ans-=ask(up[l])-ask(down[l]);
87             ++l;
88         }
89         res[t.id]=ans;
90     }
91     for(int i=0;i<m;++i)
92         printf("%d\n",res[i]);
93     return 0;
94 }