bzoj5016 [Snoi2017]一个简单的询问

题目

首先，可以确定的是，这一定是一道数据结构的题，但是，题目询问的式子过于复杂，我们要先考虑化简一下。

首先，吧get(l,r,x)变成get(r,x)−get(l−1,x)是非常显然的，新的get(n,x)，表示1…n有几个x

get(l1,r1,x)get(l2,r2,x)=(get(r1,x)−get(l1−1,x))(get(r2,x)−get(l2−1,x))
用ans(l,r,x)表示get(l,x)get(r,x)，然后，我们可以发现，这个东西是可以用莫队求解的，这样，我们只要把每一个询问拆成四个，然后就可以算了。

#include
#define N 50000
using namespace std;
typedef long long ll;
int n,m,siz,v[N+5],cl[N+5],cr[N+5],tot;
ll ans[N+5],now;
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    int x=0,b=1;
    char c=nc();
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
}
inline void write(ll x)
{
    if(x==0)putchar('0');
    else
    {
        char buf[25];
        int len=0;
        if(x<0)putchar('-'),x=-x;
        while(x)buf[++len]=x%10+'0',x/=10;
        for(int i=len;i>=1;i--)putchar(buf[i]);
    }
    putchar('\n');
}
struct data{
    int l,r,id,flg;
    data(){}
    data(int a,int b,int c,int d){l=min(a,b),r=max(a,b),flg=c,id=d;};
    bool operator <(const data & A)const{
        return ((l-1)/siz==(A.l-1)/siz?r1)/siz<(A.l-1)/siz);
    }
}a[N*4+5];
int main()
{
    freopen("in.txt","r",stdin);
    int x1,x2,y1,y2,l,r;
    n=read();siz=sqrt(n);
    for(int i=1;i<=n;i++)v[i]=read();
    m=read();l=r=0;
    for(int i=1;i<=m;i++)
    {
        x1=read(),y1=read(),x2=read(),y2=read();
        a[++tot]=data(y1,y2,1,i);
        if(x1>1)a[++tot]=data(x1-1,y2,-1,i);
        if(x2>1)a[++tot]=data(y1,x2-1,-1,i);
        if(x1>1&&x2>1)a[++tot]=data(x1-1,x2-1,1,i);
    }
    sort(a+1,a+tot+1);
    for(int i=1;i<=tot;i++)
    {
        while(lwhile(rwhile(l>a[i].l)cl[v[l]]--,now-=cr[v[l]],l--;
        while(r>a[i].r)cr[v[r]]--,now-=cl[v[r]],r--;
        ans[a[i].id]+=(ll)a[i].flg*now;
    }
    for(int i=1;i<=m;i++)write(ans[i]);
    return 0;
}