洛谷 P1972 [SDOI2009]HH的项链

洛谷 P1972 [SDOI2009]HH的项链

题目描述

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

输入格式：

 

第一行：一个整数N，表示项链的长度。

第二行：N 个整数，表示依次表示项链中贝壳的编号（编号为0 到1000000 之间的整数）。

第三行：一个整数M，表示HH 询问的个数。

接下来M 行：每行两个整数，L 和R（1 ≤ L ≤ R ≤ N），表示询问的区间。

 

输出格式：

 

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

 

 

输入样例#1： 

6
1 2 3 4 3 5
3
1 2
3 5
2 6

输出样例#1： 

2
2
4

说明

对于100%的数据，N <= 500000，M <= 200000。

题解：

莫队算法入门题之一。

不赘述，（见我的博客小Z的袜子）

哪位 dalao 能帮我看看下面的代码怎么优化？洛谷上 T 掉了2个点， BZOJ 上也 T 掉了。感谢了！！！

#include<bits/stdc++.h>
using namespace std;
const int N=500005,M=200005;
int n,m,block,col[N],belong[N],ans[M],sum,num[N*2];
struct node{
    int l,r,id;
}q[M];
inline int read()
{
    int x=0,f=1; char ch=getchar();
    while (!isdigit(ch))
      f=(ch=='-')?-f:f,ch=getchar();
    while (isdigit(ch))
      x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*f;
}
bool cmp(node a,node b)
{
    return belong[a.l]==belong[b.l]?(belong[a.r]==belong[b.r]?a.id<b.id:a.r<b.r):a.l<b.l;
}
void revise(int x,int w)
{
    num[col[x]]+=w; 
    if (w>0) sum+=(num[col[x]]==1);
    if (w<0) sum-=(num[col[x]]==0);
}
int main()
{
    n=read();
    block=sqrt(n);
    for (int i=1; i<=n; i++)
    {
        scanf("%d",&col[i]);
        belong[i]=i/block+1;
    }
    m=read();
    for (int i=1; i<=m; i++)
    {
        int x,y;
        x=read(),y=read();
        q[i].l=x,q[i].r=y,q[i].id=i;
    }
    sort(q+1,q+1+m,cmp);
    int zl=0,zr=0;
    memset(num,0,sizeof(num)); sum=0;
    for (int i=1; i<=m; i++)
    {
        while (zl<q[i].l) revise(zl,-1),zl++;
        while (zl>q[i].l) revise(zl-1,1),zl--;
        while (zr<q[i].r) revise(zr+1,1),zr++;
        while (zr>q[i].r) revise(zr,-1),zr--;
        ans[q[i].id]=sum;
    }
    for (int i=1; i<=m; i++)
      printf("%d
",ans[i]);
    return 0;
}

加油加油加油！！！ fighting fighting fighting !!!