题意:n个数字,m次查询,每次给你一个k,输出区间极差大于k的区间个数
分析:众所周知ST表可以做到O(nlogn)预处理出区间极值,O(1)回答每次询问。又因为区间极差是非递减的,故用尺取法确定最小的合法区间后加上之后的范围即可。
代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int logn=21;
const int maxn=1e5+5;
int f1[maxn][logn],f2[maxn][logn],Logn[maxn];
int n,m;
void pre()
{
Logn[1]=0;
for(int i=2;i<maxn;i++)
{
Logn[i]=Logn[i/2]+1;
}
for(int j=1;j<=logn;j++)
{
for(int i=1;i+(1<<j)-1<=n;i++)
{
f1[i][j]=max(f1[i][j-1],f1[i+(1<<(j-1))][j-1]);
f2[i][j]=min(f2[i][j-1],f2[i+(1<<(j-1))][j-1]);
}
}
}
int askmax(int l,int r)
{
int s=Logn[r-l+1];
return max(f1[l][s],f1[r-(1<<s)+1][s]);
}
int askmin(int l,int r)
{
int s=Logn[r-l+1];
return min(f2[l][s],f2[r-(1<<s)+1][s]);
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n;i++)
{
cin>>f1[i][0];
f2[i][0]=f1[i][0];
}
pre();
while(m--)
{
int k;
cin>>k;
int pos=1;
ll ans=0;
for(int i=1;i<=n;i++)
{
while(pos<=n&&askmax(i,pos)-askmin(i,pos)<=k) pos++;
if(pos==n+1) break;
ans+=n-pos+1;
}
cout<<ans<<endl;
}
}
单调队列:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e5+5;
int a[maxn],q1[maxn],q2[maxn];
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
while(m--)
{
int k;
scanf("%d",&k);
int t1=0,t2=0,h1=1,h2=1,lim=1;
ll ans=0;
for(int i=1;i<=n;i++)
{
while(h1<=t1&&a[q1[t1]]>=a[i]) t1--;//单调递增队列
while(h2<=t2&&a[q2[t2]]<=a[i]) t2--;//单调递减队列
q1[++t1]=q2[++t2]=i;//q对应队列中的编号
while(a[q2[h2]]-a[q1[h1]]>k)
{
ans+=n-i+1;
lim++;//每次进来最大编号++ 只有一个值的队列不应++
if(q1[h1]<lim) h1++;//单调队列每次只有一个队列有多个值
if(q2[h2]<lim) h2++;//所以每次其实h1,h2只有一个会变化
}
}
cout<<ans<<endl;
}
}
/*
5 3
1 2 4 2 1
1
3+3+2=8
*/