浅谈莫队算法

首先来看一道例题:

Description
有n个数字,以及m个查询。每次查询的格式是L,r,求L~r(左右包含)这个区间内有多少个不同的数?
1< n,m<=50000,1<=L<r<=n,数列中元素大小<= \(10^9\)

思路1:

自然而然想到暴力,对每次询问L~r枚举区间,考虑数列中元素取值范围较大,先要进行离散化处理。
时间复杂度:O(nm)
代码略

路2:

线段树或树状数组处理一下。(代码复杂度相对而言较高)。

#include<cstdio>
#include<algorithm>
#define lowbit(x) ((x)&(-(x)))
using namespace std;
int n,m;
int b[500010],c[500010],last[1000010],ans[500010];
struct node{int x,y,id;} a[500010];
bool cmp(node x,node y)
{
    return x.y==y.y?x.x<y.x:x.y<y.y;
}
void add(int x,int y)
{
    while(x<=n)
    {
        c[x]+=y;
        x+=lowbit(x);
    }
}
int getsum(int x)
{
    int sum=0;
    while(x)
    {
        sum+=c[x];
        x-=lowbit(x);
    }
    return sum;
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&b[i]);
    scanf("%d",&m);
    for(int i=1;i<=m;i++)
    {
        scanf("%d %d",&a[i].x,&a[i].y);
        a[i].id=i;
    }
    sort(a+1,a+m+1,cmp);
    int now=1;
    for(int i=1;i<=n;i++)
    {
        if(last[b[i]]) add(last[b[i]],-1);
        last[b[i]]=i;
        add(i,1);
        while(i==a[now].y&&now<=m)
        {
            ans[a[now].id]=getsum(a[now].y)-getsum(a[now].x-1);
            now++;
        }
        if(now==m+1) break;
    }
    for(int i=1;i<=m;i++)
        printf("%d\n",ans[i]);
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/tham/p/12200156.html
今日推荐