CSU 1515 Sequence(莫队算法)

Description
给出一个长度为n的序列ai,进行m次查询,每次查询区间[l,r]中满足|ai-aj|=1的(i,j)对数(l<=i < j<=r)
Input
第一行为两整数n和m表示序列长度和查询次数,第二行n个整数ai表示该序列,最后m行每行两个整数l和r表示查询区间(1<=n<=10^4,1<=m<=10^5,0<=ai< 2^31)
Output
输出每次查询的答案
Sample Input
10 10
5 5 1 3 6 3 5 7 1 7
3 4
6 8
8 9
2 8
5 7
6 7
1 9
3 10
3 10
5 6
Sample Output
0
0
0
3
1
0
4
3
3
0
Solution
莫队,给所有操作首先按左端点平方分块,然后将每个区间左端点所处块的编号为第一关键字,右端点为第二关键字排序,之后按顺序在区间间递推即可
Code

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<map>
using namespace std;
#define maxn 11111
struct node
{
    int l,r,id,ans;
}q[10*maxn];
int n,m,ans,a[maxn],pos[maxn];
map<int,int>cnt;
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];
    cnt[x]+=v;
    ans+=cnt[x+1]*v;
    if(x)ans+=cnt[x-1]*v;
}
int main()
{
    while(~scanf("%d%d",&n,&m))
    {
        cnt.clear();
        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;
}

你可能感兴趣的:(CSU 1515 Sequence(莫队算法))