bzoj 3809(莫队+分块求和)

3809: Gty的二逼妹子序列

Time Limit: 80 Sec   Memory Limit: 28 MB
Submit: 1185   Solved: 330
[ Submit][ Status][ Discuss]

Description

Autumn和Bakser又在研究Gty的妹子序列了！但他们遇到了一个难题。

对于一段妹子们，他们想让你帮忙求出这之内美丽度∈[a,b]的妹子的美丽度的种类数。

为了方便，我们规定妹子们的美丽度全都在[1,n]中。

给定一个长度为n(1<=n<=100000)的正整数序列s(1<=si<=n)，对于m(1<=m<=1000000)次询问“l,r,a,b”，每次输出sl...sr中，权值∈[a,b]的权值的种类数。

Input

第一行包括两个整数n,m(1<=n<=100000,1<=m<=1000000),表示数列s中的元素数和询问数。

第二行包括n个整数s1...sn(1<=si<=n)。

接下来m行,每行包括4个整数l,r,a,b(1<=l<=r<=n,1<=a<=b<=n),意义见题目描述。

保证涉及的所有数在C++的int内。

保证输入合法。

Output

对每个询问，单独输出一行，表示sl...sr中权值∈[a,b]的权值的种类数。

Sample Input

10 10
4 4 5 1 4 1 5 1 2 1
5 9 1 2
3 4 7 9
4 4 2 5
2 3 4 7
5 10 4 4
3 9 1 1
1 4 5 9
8 9 3 3
2 2 1 6
8 9 1 4

Sample Output

2
0
0
2
1
1
1
0
1

2

解题思路：

首先莫队，然后剩下就是如何统计颜色了，这里树状数组要超时，可以用分块（太胜了）

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cmath>
using namespace std;
int n,m,opp,lnow,rnow;
struct ss
 {
    int l,r,a,b,dui;
 }q[1000011];
int pos[100100],sum[100100],shu[100100],s[100100];
int ans[1000010];
 
inline int read()
{
    char y; int x=0,f=1; y=getchar();
    while (y<'0'||y>'9') {if (y=='-') f=-1; y=getchar();}
    while (y>='0' && y<='9') {x=x*10+int(y)-48; y=getchar();}
    return x*f;
}
 
bool cmp(ss x,ss y)
 {
    if ((x.l-1)/opp+1!=(y.l-1)/opp+1)return (x.l-1)/opp+1<(y.l-1)/opp+1;else
     return x.r<y.r;
 }
 
int query(int st,int en)
 {
    int sug=0;
    if (pos[st]==pos[en])
     {
        for (int i=st;i<=en;++i)
         if (shu[i]>=1)++sug;
        return sug;
      }
    for (int i=pos[st]+1;i<=pos[en]-1;++i)
     {
        sug+=sum[i];
      }
    for (int i=st;i<=pos[st]*opp;++i)
     if (shu[i]>=1) ++sug;
    for (int i=(pos[en]-1)*opp+1;i<=en;++i)
     if (shu[i]>=1) ++sug;
    return sug;
 }


void dfs()
 {
 
 }


int main()
{
    n=read(); m=read();
    for (int i=1;i<=n;++i) s[i]=read(),shu[i]=0; opp=sqrt(n);
    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].dui=i;
     }
    sort(q+1,q+m+1,cmp);
    for (int i=1;i<=n;++i)
     {
        pos[i]=(i-1)/opp+1; sum[pos[i]]=0;
     }
    lnow=1; rnow=n;
    for (int i=1;i<=n;++i)
     {
        ++shu[s[i]];
        if (shu[s[i]]==1) ++sum[pos[s[i]]];
     }
    for (int i=1;i<=m;++i)
     {
      if (i==1000000)
      {
      dfs();
 }
        while (lnow<q[i].l)
         {
            --shu[s[lnow]];
            if (shu[s[lnow]]==0) --sum[pos[s[lnow]]];
            lnow++;
         }
        while (lnow>q[i].l)
         {
            ++shu[s[lnow-1]];
            if (shu[s[lnow-1]]==1) ++sum[pos[s[lnow-1]]];
            --lnow;
         }
        while (rnow<q[i].r)
         {
            ++shu[s[rnow+1]];
            if (shu[s[rnow+1]]==1) ++sum[pos[s[rnow+1]]];
            ++rnow;
         }
        while (rnow>q[i].r)
         {
            --shu[s[rnow]];
            if (shu[s[rnow]]==0) --sum[pos[s[rnow]]];
            --rnow;
         }
        ans[q[i].dui]=query(q[i].a,q[i].b);
     }
    for (int i=1;i<=m;++i)
     {
        cout<<ans[i]<<endl;
     }
}