[bzoj5016][Snoi2017]一个简单的询问

来自FallDream的博客，未经允许，请勿转载，谢谢。

---

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

 

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

n,q<=50000

Ans(l1,r1,l2,r2)=Ans(1,r1,1,r2)-Ans(1,l1-1,1,r2)-Ans(1,r1,1,l2-1)+Ans(1,l1-1,1,l2-1)

然后就可以莫队了

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cmath>
#define MN 50000
using namespace std;
inline int read()
{
    int x=,f=;char ch=getchar();
    while(ch<''||ch>''){if(ch=='-')f=-;ch=getchar();}
    while(ch>=''&&ch<=''){x=x*+ch-'';ch=getchar();}
    return x*f;
}
struct ques{int l,r,id,ad;}q[MN*+];
int n,block[MN+],size,a[MN+],num1[MN+],num2[MN+],m,tot=,L=,R=;
long long res=,Ans[MN+];
bool cmp(const ques&a,const ques&b){return block[a.l]==block[b.l]?a.r<b.r:a.l<b.l;}
int main()
{
    n=read();size=sqrt(n);
    for(int i=;i<=n;++i) block[i]=(i-)/size+;
    for(int i=;i<=n;++i) a[i]=read();
    m=read();
    for(int i=;i<=m;++i)
    {
        int l1=read(),r1=read(),l2=read(),r2=read();
        q[++tot]=(ques){r1,r2,i,};
        q[++tot]=(ques){r1,l2-,i,-};
        q[++tot]=(ques){l1-,r2,i,-};
        q[++tot]=(ques){l1-,l2-,i,};
    }
    sort(q+,q+tot+,cmp);
    for(int i=;i<=tot;++i)
    {
        while(L<q[i].l) ++L,res+=num2[a[L]],++num1[a[L]];
        while(L>q[i].l) res-=num2[a[L]],--num1[a[L]],--L;
        while(R<q[i].r) ++R,res+=num1[a[R]],++num2[a[R]];
        while(R>q[i].r) res-=num1[a[R]],--num2[a[R]],--R;
        Ans[q[i].id]+=q[i].ad*res;
    }
    for(int i=;i<=m;++i) printf("%lld\n",Ans[i]);
    return ;
}