题目链接:Click here~~
题意:
给一个长度为 n 的序列,且序列元素为 1 ~ n 的全排列。Q 次询问,每次询问区间 [l,r] 有多少段连续的数字(可重新排列)。
解题思路:
继续刷啦啦啦。
首先可以在 O(n) 的时间内处理出所有关于区间 [1,r] 的询问。
考虑每次将区间右端点扩张,新出现的数对上次答案产生的影响 delta ,只有 -1,0,1 三种,递推过去就得到了 ans of [1,r]。
但这里我们不用递推,而是用树状数组维护每个位置的 delta 值,位置的前缀和表示它对应的答案。
再试着考虑如果将区间的左端点收缩([2,r]、[3,r] ……),能不能在很快的时间内维护答案。
不妨设左端点的数为 v ,它的相邻数为 v1、v2(v1 or v2 = v + 1 or v - 1),且出现顺序也为 v、v1、v2。
在删除 v 之前,一定有 delta_v = 1 , delta_v1 = 0 or -1 , delta_v2 = 0 or -1。而删除 v 也仅仅会影响这两个数的 delta。
分情况考虑可以得出 不管 delta_vx 是几,删除 v 都会使它们 +1,所以我们就有办法用树状数组在 O(log n) 时间内维护了。
而如果出现顺序为 v1、v、v2,delta_v2 还是会加 1,而对于 delta_v1 的加 1 不会对答案有影响。v1、v2、v 同样如此,所以不用特判可以放心操作了。
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int N = 1e5 + 5;
struct QAQ
{
int l,r,id;
void read(int i){
scanf("%d%d",&l,&r);
id = i;
}
bool operator < (const QAQ& B) const{
return l < B.l;
}
}query[N];
int a[N],c[N],pos[N],ans[N];
inline int lowbit(int x){
return x & -x;
}
void add(int loc,int val){
while(loc < N){
c[loc] += val;
loc += lowbit(loc);
}
}
int sum(int loc){
int ret = 0;
while(loc){
ret += c[loc];
loc -= lowbit(loc);
}
return ret;
}
int sum(int a,int b){
return sum(b) - sum(a-1);
}
int main()
{
int T,n,Q;
scanf("%d",&T);
while(T--)
{
memset(pos,0,sizeof(pos));
memset(c,0,sizeof(c));
scanf("%d%d",&n,&Q);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
int num = a[i];
if(!pos[num-1] && !pos[num+1])
add(i,1);
else if(pos[num-1] && pos[num+1])
add(i,-1);
pos[num] = i;
}
for(int i=0;i<Q;i++)
query[i].read(i);
sort(query,query+Q);
int i = 1;
for(int j=0;j<Q;j++)
{
while(i < query[j].l)
{
int v = a[i];
if(v != n)
add(pos[v+1],1);
if(v != 1)
add(pos[v-1],1);
i++;
}
ans[ query[j].id ] = sum(query[j].l,query[j].r);
}
for(int i=0;i<Q;i++)
printf("%d\n",ans[i]);
}
return 0;
}