[bzoj1878][SDOI2009][HH的项链] (莫队算法)

Description

HH有一串由各种漂亮的贝壳组成的项链。HH相信不同的贝壳会带来好运，所以每次散步完后，他都会随意取出一段贝壳，思考它们所表达的含义。HH不断地收集新的贝壳，因此，他的项链变得越来越长。有一天，他突然提出了一个问题：某一段贝壳中，包含了多少种不同的贝壳？这个问题很难回答。。。因为项链实在是太长了。于是，他只好求助睿智的你，来解决这个问题。

Input

第一行：一个整数N，表示项链的长度。第二行：N个整数，表示依次表示项链中贝壳的编号（编号为0到1000000之间的整数）。第三行：一个整数M，表示HH询问的个数。接下来M行：每行两个整数，L和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

Output

M行，每行一个整数，依次表示询问对应的答案。

Sample Input

Sample Output

HINT

对于20%的数据，N ≤ 100，M ≤ 1000；
对于40%的数据，N ≤ 3000，M ≤ 200000；
对于100%的数据，N ≤ 50000，M ≤ 200000。

Solution

分块大小开n^(2/3)就跪了，乖乖开n^(1/2)

//Kaiba_Seto 20170116

//orz cjkmao

#include <math.h>

#include <stdio.h>

#include <string.h>

#include <algorithm>

#define MaxN 50010

#define MaxM 200010

#define MaxS 1000010

#define MaxBuf 1<<22

#define RG register

#define inline __inline__ __attribute__((always_inline))

#define Blue() (S == T&&(T=(S=B)+fread(B,1,MaxBuf,stdin),S == T)?0:*S++)

char B[MaxBuf],*S=B,*T=B;

inline void Rin(RG int &x) {

    x=;RG int c=Blue(),f=;

    for(; c < ||c > ; c=Blue())

        if(c == )f=-;

    for(; c > &&c < ; c=Blue())

        x=(x<<)+(x<<)+c-;

    x*=f; }

int n,m,a[MaxN],cnt[MaxS],block_size,ans[MaxM];

struct Request{

    int l,r,id;

    bool operator < (const Request &other)const {

        if(l/block_size == other.l/block_size)

            return r/block_size < other.r/block_size;

        return l/block_size < other.l/block_size; } }Q[MaxM];

inline void extend(RG int &tmp,RG int pos,RG int dir) {

    if(dir == ) {

        if(++cnt[a[pos]] == )tmp++; }

    else

        if(--cnt[a[pos]] == )tmp--; }

inline void block_solve() {

    for(RG int l=,r=,ans=,i=; i<=m; i++) {

        while(l < Q[i].l)extend(ans,l++,-);

        while(l > Q[i].l)extend(ans,--l,);

        while(r > Q[i].r)extend(ans,r--,-);

        while(r < Q[i].r)extend(ans,++r,);

        :: ans[Q[i].id]=ans; } }

int main() {

    Rin(n);

    for(RG int i=; i<=n; i++)

        Rin(a[i]);

    Rin(m);

    for(RG int i=; i<=m; i++)

        Rin(Q[i].l),Rin(Q[i].r),Q[i].id=i;

    //block_size=floor(pow(n,2.0/3)+1);

    block_size=static_cast<int>(sqrt(n));

    std::sort(Q+,Q++m);

    block_solve();

    for(RG int i=; i<=m; i++)

        printf("%d\n",ans[i]);

    return ; }

秒客网