【HDU 4638 Group】 莫队算法

HDU4638
题意就是给你一个长度为n的序列，每次查询给一个区间，问区间可被分成多少个连续的段
例如 1,2,4,5被分为 [1,2],[4,5]两段。
我们可以考虑添加一个数对答案的影响，添加一个数的时候，如果他左右的数字都存在，显然段数-1
若左右存在某一个，段数不变
若左右均不存在，则段数+1
删除一个数的时候类似，自己推导一下就可以了。
代码

#include<stdio.h>
#include<math.h>
#include<iostream>
#include<algorithm>
#include<set>
using namespace std;
#define dbg(x) cout<<#x<<" = "<<x<<endl;
const int maxn = 1e5+5;
int a[maxn];
int pos[maxn];
int ans[maxn];
int vis[maxn];
struct data
{
    int l,r,id;
}Q[maxn];
bool cmp(const data &a,const data &b)
{
    if(pos[a.l]==pos[b.l]) return a.r<b.r;
    return pos[a.l]<pos[b.l];
}
int L,R,Ans;
void add(int i)
{
    int id=a[i];
    Ans-=(vis[id-1]+vis[id+1]-1);
    vis[id]=1;
}
void del(int i)
{
    int id=a[i];
    Ans+=(vis[id-1]+vis[id+1]-1);
    vis[id]=0;
}
int main()
{
    int n,m;
    int T;
    scanf("%d",&T);
    while(T--)
    {
        L=1,R=0,Ans=0;
        scanf("%d%d",&n,&m);
        for(int i=0;i<=2*n;i++)
        {
            vis[i]=0;
        }
        int sz=sqrt(n);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            pos[i]=i/sz;
        }
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&Q[i].l,&Q[i].r);
            Q[i].id=i;
        }
        sort(Q+1,Q+1+m,cmp);
        for(int i=1;i<=m;i++)
        {
            while(R<Q[i].r)
            {
                R++;
                add(R);
            }
             while(L>Q[i].l)
            {
                L--;
                add(L);
            }
            while(R>Q[i].r)
            {
                del(R);
                R--;
            }
            while(L<Q[i].l)
            {
                del(L);
                L++;
            }
            ans[Q[i].id]=Ans;
        }
        for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
    }
    return 0;
}

友情推荐莫队算法专题链接