bzoj 3809: Gty的二逼妹子序列 分块+莫队算法

题意

给出n个数m次询问，每次询问[a,b]之间的颜色在区间[l,r]之间出现了多少种。
n<=100000,m<=1000000

分析

看到28M的内存整个人都不好了啊。
树套树什么的就别想了~~
这题很容易想到莫队+树状数组，那样的话复杂度是$O(n\sqrt nlogn)$
还有一种离奇的分块做法，就是对权值分块，那么修改就是$O(1)$，查询就是$O(n\sqrt n)$
那么总复杂度就是$O(n\sqrt n)$了

好劲啊居然跑了40s+，不过我看到居然还有跑了80s+的……
卡常出奇迹啊~~

代码

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

int pos[N],sta[M],end[M],t[N],sum[M],n,m,a[N],block;
struct que{int l,r,a,b,ans,id;}q[Q];

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;
}

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

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

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

int query(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+=sum[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 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].ans=query(q[i].a,q[i].b);
    }
}

int main()
{
    //freopen("3809.in","r",stdin);freopen("test.out","w",stdout);
    n=read();m=read();
    block=sqrt(n);
    for (int i=1;i<=n;i++)
    {
        a[i]=read();
        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++)
    {
        q[i].l=read();q[i].r=read();q[i].a=read();q[i].b=read();
        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\n",q[i].ans);
    return 0;
}