King of Range

原题链接
考察:双指针
思路:
  两个单调队列,一个存最大值,一个存最小值.当差值\(>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;
}
上一篇:Never late, never early


下一篇:牛客多校2021(五)K.King of Range(ST表、双指针)