bzoj3809: Gty的二逼妹子序列

链接

　　http://www.lydsy.com/JudgeOnline/problem.php?id=3809

题解

　　先说一种很显然的做法，大体框架肯定是莫队，然后每来一种颜色就处理下，如果是新出现的就在BIT当中把这个位置+1，如果这种颜色消失了就在BIT中对应位置-1。
　　这样的时间复杂度是$O(N\sqrt Nlog_2N)$
　　我交了一次发现超时了。
　　那怎么办？
　　观察下，我们查询和修改的复杂度都是$O(logN)$，但是显然查询的次数要远远小于修改次数，因此考虑牺牲查询的时间使修改变快。
　　都想到这里了，显然接下来就要上分块了。用权值分块代替BIT，每个块内记录出现了多少种颜色，用一个桶辅助修改，使修改的复杂度降到$O(1)$，查询则变成$O(\sqrt N)$。
　　总的时间复杂度$O(Nlog_2N+N\sqrt N+M\sqrt N)$

代码

//莫队+分块 
#include <cstdio>
#include <algorithm>
#include <ctime>
#define maxn 100010
using namespace std;
int cnt[maxn], num[maxn*10], q[maxn*10][4], size, N, M, col[maxn],
    ans[maxn*10], pos[maxn], block[maxn];
inline bool cmp(int a, int b)
{return q[a][0]>>size==q[b][0]>>size?q[a][1]<q[b][1]:q[a][0]>>size<q[b][0]>>size;}
inline int read(int x=0)
{
    char c=getchar();
    while(c<48 or c>57)c=getchar();
    while(c>=48 and c<=57)x=(x<<1)+(x<<3)+c-48,c=getchar();
    return x;
}
void init()
{
    int i, j;
    N=read(), M=read();
    for(i=1;i<=N;i++)col[i]=read();
    for(i=1;i<=M;i++)for(j=0;j<4;j++)q[i][j]=read();
    for(i=1;i<=M;i++)num[i]=i;
    for(size=0;(1<<size)*(1<<size)<N;size++);
    sort(num+1,num+M+1,cmp);
}
inline void add(int c){if(++cnt[c]==1)block[c>>size]++;}
inline void del(int c){if(--cnt[c]==0)block[c>>size]--;}
inline int quiry(int l, int r)
{
    int i, ans=0;
    for(i=(l>>size)+1;i<(r>>size);i++)ans+=block[i];
    for(i=l;i<=r and i>>size==l>>size;i++)if(cnt[i])ans++;
    if((l>>size)^(r>>size))for(i=r;i>>size==r>>size;i--)if(cnt[i])ans++;
    return ans;
}
void work()
{
    int i, l, r, x, y;
    l=0,r=2;add(col[1]);
    for(i=1;i<=M;i++)
    {
        x=q[num[i]][0], y=q[num[i]][1];
        for(;l>=x;l--)add(col[l]);
        for(;l<x-1;)l++,del(col[l]);
        for(;r<=y;r++)add(col[r]);
        for(;r>y+1;)r--,del(col[r]);
        ans[num[i]]=quiry(q[num[i]][2],q[num[i]][3]);
    }
    for(i=1;i<=M;i++)printf("%d\n",ans[i]);
}
int main()
{
    init();
    work();
    return 0;
}