hdu 4638 Group （树状数组+离线）

来源：互联网发布：五十知天命六十编辑：程序博客网时间：2024/05/29 16:30

题意：给你一串数，然后对于每次询问l和r，问你在[l,r]中有多少段连续的数

思路：树状数组，离线。每向后添加一个i位置的数a[i]，先直接在这个位置+1，表示出现一个新组，然后看前面是否出现a[i]-1和a[i]+1，若出现a[i]-1，则在pos[a[i]-1]位置-1，a[i]+1同理。因为此时数段被合并起来了，a[i]就代表了它们的状态。将所有的询问按r从小到大排序。然后枚举i，树状数组维护的是第i个数之前那些-1,0,1的和。

#include<iostream>#include<algorithm>#include<cstring>#include<string>#include<cstdio>using namespace std;#define lowbit(x) ((x)&(-(x)))const int maxn=100001;struct point{    int l,r,id;    bool operator < (const point & p) const    {        return r<p.r;    }};int T,n,m;int a[maxn],c[maxn],pos[maxn],ans[maxn];bool mark[maxn];point q[maxn];void update(int *a,int x,int w){    while(x<=n)    {        a[x]+=w;        x+=lowbit(x);    }}int sum(int *a,int x){    int ans=0;    while(x>0)    {        ans+=a[x];        x-=lowbit(x);    }    return ans;}int main(){    scanf("%d",&T);    while(T--)    {        scanf("%d%d",&n,&m);        for(int i=1;i<=n;i++)        {            scanf("%d",&a[i]);            pos[a[i]]=i;        }        for(int i=1;i<=m;i++)        {            scanf("%d%d",&q[i].l,&q[i].r);            q[i].id=i;        }        sort(q+1,q+m+1);        memset(mark,0,sizeof(mark));        memset(c,0,sizeof(c));        int j=1;        for(int i=1;i<=n;i++)        {            mark[a[i]]=1;            update(c,pos[a[i]],1);            if(a[i]-1>0 && mark[a[i]-1])                update(c,pos[a[i]-1],-1);            if(a[i]+1<=n && mark[a[i]+1])                update(c,pos[a[i]+1],-1);            while(q[j].r==i)            {                ans[q[j].id]=sum(c,q[j].r)-sum(c,q[j].l-1);                j++;            }        }        for(int i=1;i<=m;i++)            printf("%d\n",ans[i]);    }    return 0;}