bzoj5016 [Snoi2017]一个简单的询问

Description

给你一个长度为N的序列ai，1≤i≤N和q组询问，每组询问读入l1,r1,l2,r2，需输出 

get(l,r,x)表示计算区间[l,r]中，数字x出现了多少次。

Input

第一行，一个数字N，表示序列长度。

第二行，N个数字，表示a1～aN

第三行，一个数字Q，表示询问个数。

第4～Q+3行，每行四个数字l1,r1,l2,r2，表示询问。

N,Q≤50000

N1≤ai≤N

1≤l1≤r1≤N

1≤l2≤r2≤N

注意：答案有可能超过int的最大值

Output

对于每组询问，输出一行一个数字，表示答案 

Sample Input

5
1 1 1 1 1
2
1 2 3 4
1 1 4 4

Sample Output

4
1

正解：莫队算法。

把每个询问看成一个矩形，拆成$4$个二维前缀和类型的子矩形。

然后直接用莫队就可以得出答案了，注意这个莫队的姿势和普通莫队有点不一样。。

#include <bits/stdc++.h>
#define il inline
#define RG register
#define ll long long
#define N (200010)

using namespace std;

struct data{ int i,l,r,v; }q[N];

int a[N],bl[N],cntl[N],cntr[N],Q,n,cnt,block;
ll ans[N];

il int gi(){
  RG int x=0,q=1; RG char ch=getchar();
  while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
  if (ch=='-') q=-1,ch=getchar();
  while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar();
  return q*x;
}

il int cmp(const data &a,const data &b){
  if (bl[a.l]==bl[b.l]) return a.r<b.r; return bl[a.l]<bl[b.l];
}

int main(){
#ifndef ONLINE_JUDGE
  freopen("query.in","r",stdin);
  freopen("query.out","w",stdout);
#endif
  n=gi(),block=sqrt(n);
  for (RG int i=1;i<=n;++i) a[i]=gi(),bl[i]=(i-1)/block+1; Q=gi();
  for (RG int i=1,l1,r1,l2,r2;i<=Q;++i){
    l1=gi(),r1=gi(),l2=gi(),r2=gi();
    q[++cnt]=(data){i,r1,r2,1};
    q[++cnt]=(data){i,l1-1,r2,-1};
    q[++cnt]=(data){i,r1,l2-1,-1};
    q[++cnt]=(data){i,l1-1,l2-1,1};
  }
  for (RG int i=1;i<=cnt;++i) if (q[i].l>q[i].r) swap(q[i].l,q[i].r);
  sort(q+1,q+cnt+1,cmp); RG int L=0,R=0; RG ll Ans=0;
  for (RG int i=1;i<=cnt;++i){
    if (!q[i].l || !q[i].r) continue;
    while (L>q[i].l) Ans-=cntr[a[L]],--cntl[a[L--]];
    while (R<q[i].r) ++cntr[a[++R]],Ans+=cntl[a[R]];
    while (L<q[i].l) ++cntl[a[++L]],Ans+=cntr[a[L]];
    while (R>q[i].r) Ans-=cntl[a[R]],--cntr[a[R--]];
    ans[q[i].i]+=1LL*q[i].v*Ans;
  }
  for (RG int i=1;i<=Q;++i) printf("%lld
",ans[i]); return 0;
}