题意为询问一段区间里的数能组成多少段连续的数。先考虑从左往右一个数一个数添加,考虑当前添加了i - 1个数的答案是x,那么添加完i个数后的答案是多少?可以看出,是根据a[i]-1和a[i]+1是否已经添加而定的,如果a[i]-1或者a[i]+1已经添加一个,则段数不变,如果都没添加则段数加1,如果都添加了则段数减1。设v[i]为加入第i个数后的改变量,那么加到第x数时的段数就是sum{v[i]} (1<=i<=x}。仔细想想,若删除某个数,那么这个数两端的数的改变量也会跟着改变,这样一段区间的数构成的段数就还是他们的v值的和。将询问离线处理,按左端点排序后扫描一遍,左边删除,右边插入,查询就是求区间和。
树状数组+离线查询
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 100010;
int c[maxn],pos[maxn],a[maxn],ans[maxn];
int n,m;
void updata(int p,int v)
{
for(int i=p;i<=n;i+=i&(-i))
c[i]+=v;
}
int sum(int p)
{
int res=0;
for(int i=p;i>0;i-=i&(-i))
res+=c[i];
return res;
}
struct Q
{
int l,r,id;
bool operator < (const Q a)const
{
return r<a.r;
}
}q[maxn];
int main()
{
int T;
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=0;i<m;i++)
scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;
sort(q,q+m);
memset(c,0,sizeof(c));
for(int i=1,k=0;i<=n;i++)
{
updata(i,1);
if(a[i]>1&&pos[a[i]-1]<i)
updata(pos[a[i]-1],-1);
if(a[i]<n&&pos[a[i]+1]<i)
updata(pos[a[i]+1],-1);
while(q[k].r==i&&k<m)
{
ans[q[k].id]=sum(q[k].r)-sum(q[k].l-1);
k++;
}
}
for(int i=0;i<m;i++)
printf("%d\n",ans[i]);
}
}