HDU 4638 Group(莫队算法)

Description
给出一个由1~n组成的序列,m次查询,每次查询区间[l,r]中有多少连续的数字段
Input
第一行为一整数T表示用例组数,每组用例第一行为两整数n和m分别表示序列长度和查询次数,第二行为n个整数表示该序列,最后k行每行两个整数l和r表示查询区间
Output
对于每次查询,输出查询答案
Sample Input
1
5 2
3 1 2 5 4
1 5
2 4
Sample Output
1
2
Solution
由[l,r]推这里写图片描述可以在O(1)内完成,所以可以用莫队算法离线处理
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<map>
using namespace std;
#define maxn 111111
struct node
{
    int l,r,id,ans;
}q[maxn];
int T,n,m,ans,a[maxn],pos[maxn],cnt[maxn];
int cmp1(node x,node y)
{
    if(pos[x.l]!=pos[y.l])return x.l<y.l;
    return x.r<y.r;
}
int cmp2(node x,node y)
{
    return x.id<y.id;
}
void update(int x,int v)
{
    x=a[x];
    if(!cnt[x-1]&&!cnt[x+1])ans+=v;
    if(cnt[x-1]&&cnt[x+1])ans-=v;
    cnt[x]+=v;
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        memset(cnt,0,sizeof(cnt));
        int mm=(int)sqrt(n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            pos[i]=(i-1)/mm+1;
        }
        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i;
        }
        sort(q,q+m,cmp1);
        ans=0;
        int l=1,r=0;
        for(int i=0;i<m;i++)
        {
            while(r<q[i].r)update(r+1,1),r++;
            while(r>q[i].r)update(r,-1),r--;
            while(l<q[i].l)update(l,-1),l++;
            while(l>q[i].l)update(l-1,1),l--;
            q[i].ans=ans;
        }
        sort(q,q+m,cmp2);
        for(int i=0;i<m;i++)printf("%d\n",q[i].ans);
    }
    return 0;
}

你可能感兴趣的:(HDU 4638 Group(莫队算法))