原题链接
考察:双指针
思路:
两个单调队列,一个存最大值,一个存最小值.当差值\(>k\)时开始计数.这时需要在设置一个指针\(l\),将最左边的队列元素去掉.计算的时候....不要两头计算,确定左边计算右边...........
Code
#include <iostream>
#include <cstring>
using namespace std;
typedef long long LL;
const int N = 100010;
int n,m,a[N],qmax[N],qmin[N],k;
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) scanf("%d",&a[i]);
while(m--)
{
scanf("%d",&k);
LL ans = 0;
int h1 = 0,t1 = -1,h2 = 0,t2 = -1,l = 0;
for(int i=1;i<=n;i++)
{
while(h1<=t1&&a[qmax[t1]]<a[i]) t1--;
while(h2<=t2&&a[qmin[t2]]>a[i]) t2--;
qmax[++t1] = i;
qmin[++t2] = i;
while(h1<=t1&&h2<=t2&&a[qmax[h1]]-a[qmin[h2]]>k)
{
ans+=n-i+1;
++l;
if(qmax[h1]==l) ++h1;
if(qmin[h2]==l) ++h2;
}
}
printf("%lld\n",ans);
}
return 0;
}