bzoj 5016: [Snoi2017]一个简单的询问 莫队算法+分块

题意

给你一个长度为N的序列ai，1≤i≤N和q组询问，每组询问读入l1,r1,l2,r2，需输出
get(l,r,x)表示计算区间[l,r]中，数字x出现了多少次。
N,Q≤50000

分析

我们要求的实际上就是区间[l1,r1]中的每个数在区间[l2,r2]中出现次数的和。
考虑按位置分块，预处理f[i,j]表示第i块中的每个数在[1,j]中出现次数的和。
询问的时候，整块的可以O(1)查询，那么多出来的怎么办呢？
如果拿可持久化线段树来查询的话，复杂度要多一个$\sqrt{log}$。
注意到现在的瓶颈在于如何快速求[l2,r2]中某个数的出现次数，那么只要把询问按照[l2,r2]莫队一下就好了。
时间复杂度$O(n\sqrt n)$

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;

typedef long long LL;

const int N=50005;
const int M=250;

int n,tot,bel[N],B,sta[M],end[M],t[N],a[N];
LL f[M][N],ans[N];
struct data{int l1,r1,l2,r2,id;}q[N];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void prework()
{
    B=sqrt(n);
    for (int i=1;i<=n;i++)
    {
        bel[i]=(i+B-1)/B;
        if (!sta[bel[i]]) sta[bel[i]]=i;
        end[bel[i]]=i;
    }
    for (int i=1;i<=bel[n];i++)
    {
        for (int j=sta[i];j<=end[i];j++) t[a[j]]++;
        for (int j=1;j<=n;j++) f[i][j]=f[i][j-1]+t[a[j]];
        for (int j=sta[i];j<=end[i];j++) t[a[j]]=0;
    }
}

bool cmp(data a,data b)
{
    return bel[a.l2]<bel[b.l2]||bel[a.l2]==bel[b.l2]&&a.r2<b.r2;
}

LL query(int l1,int r1,int l2,int r2)
{
    LL ans=0;
    if (bel[l1]==bel[r1])
    {
        for (int i=l1;i<=r1;i++) ans+=t[a[i]];
        return ans;
    }
    for (int i=l1;i<=end[bel[l1]];i++) ans+=t[a[i]];
    for (int i=sta[bel[r1]];i<=r1;i++) ans+=t[a[i]];
    for (int i=bel[l1]+1;i<bel[r1];i++) ans+=f[i][r2]-f[i][l2-1];
    return ans;
}

void solve()
{
    for (int l=1,r=0,i=1;i<=tot;i++)
    {
        for (;r<q[i].r2;r++) t[a[r+1]]++;
        for (;l>q[i].l2;l--) t[a[l-1]]++;
        for (;r>q[i].r2;r--) t[a[r]]--;
        for (;l<q[i].l2;l++) t[a[l]]--;
        ans[q[i].id]=query(q[i].l1,q[i].r1,q[i].l2,q[i].r2);
    }
}

int main()
{
    n=read();
    for (int i=1;i<=n;i++) a[i]=read();
    prework();
    tot=read();
    for (int i=1;i<=tot;i++) q[i].l1=read(),q[i].r1=read(),q[i].l2=read(),q[i].r2=read(),q[i].id=i;
    sort(q+1,q+tot+1,cmp);
    solve();
    for (int i=1;i<=tot;i++) printf("%d\n",ans[i]);
    return 0;
}