bzoj 3236: [Ahoi2013]作业 莫队算法+分块

题意

给出一个长度为n的序列，元素范围为[1,n]。给出求个询问，格式为l r a b，表示询问[l,r]中在[a,b]范围内的数有多少个和在[a,b]范围内的数有多少种。
n<=100000 m<=1000000

分析

搞不懂那些莫队+树状数组是怎么卡过的，复杂度明明已经炸掉不知道多少了……

其实只要对权值分块即可做到$O(1)$修改，$O(\sqrt n)$询问，那么总时间复杂度就是$O(n\sqrt n)$啦。

跑了30s+，跟那些跑了接近100s的相比还是算快了吧，但第一的5s好劲啊，怎么跑出来的%%%

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<cmath>
#define N 100005
#define M 355
#define Q 1000005
using namespace std;

int n,m,t[N],pos[N],end[M],sta[M],sum1[M],sum2[M],a[N],block;
struct query{int l,r,a,b,id,ans1,ans2;}q[Q];

bool cmp(query a,query b)
{
    return pos[a.l]<pos[b.l]||pos[a.l]==pos[b.l]&&a.r<b.r;
}

bool cmpid(query a,query b)
{
    return a.id<b.id;
}

int query1(int l,int r)
{
    if (pos[l]==pos[r])
    {
        int tot=0;
        for (int i=l;i<=r;i++)
            tot+=t[i];
        return tot;
    }
    int tot=0;
    for (int i=pos[l]+1;i<pos[r];i++)
        tot+=sum2[i];
    for (int i=l;i<=end[pos[l]];i++)
        tot+=t[i];
    for (int i=sta[pos[r]];i<=r;i++)
        tot+=t[i];
    return tot;
}

int query2(int l,int r)
{
    if (pos[l]==pos[r])
    {
        int tot=0;
        for (int i=l;i<=r;i++)
            if (t[i]) tot++;
        return tot;
    }
    int tot=0;
    for (int i=pos[l]+1;i<pos[r];i++)
        tot+=sum1[i];
    for (int i=l;i<=end[pos[l]];i++)
        if (t[i]) tot++;
    for (int i=sta[pos[r]];i<=r;i++)
        if (t[i]) tot++;
    return tot;
}

void updata(int x,int y)
{
    if (y==1)
    {
        t[x]++;sum2[pos[x]]++;
        if (t[x]==1) sum1[pos[x]]++;
    }else
    {
        t[x]--;sum2[pos[x]]--;
        if (!t[x]) sum1[pos[x]]--;
    }
}

void solve()
{
    int l=1,r=1;
    updata(a[1],1);
    for (int i=1;i<=m;i++)
    {
        for (;r<q[i].r;r++)
            updata(a[r+1],1);
        for (;l>q[i].l;l--)
            updata(a[l-1],1);
        for (;r>q[i].r;r--)
            updata(a[r],-1);
        for (;l<q[i].l;l++)
            updata(a[l],-1);
        q[i].ans1=query1(q[i].a,q[i].b);
        q[i].ans2=query2(q[i].a,q[i].b);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    block=sqrt(n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        pos[i]=(i+block-1)/block;
        if (!sta[pos[i]]) sta[pos[i]]=i;
        end[pos[i]]=i;
    }
    for (int i=1;i<=m;i++)
    {
        scanf("%d%d%d%d",&q[i].l,&q[i].r,&q[i].a,&q[i].b);
        q[i].id=i;
    }
    sort(q+1,q+m+1,cmp);
    solve();
    sort(q+1,q+m+1,cmpid);
    for (int i=1;i<=m;i++)
        printf("%d %d\n",q[i].ans1,q[i].ans2);
    return 0;
}