hdu5289(双端队列的应用)

解题思路:这里使用双端队列来维护最大值和最小值,然后结合尺取法来解题。其中low维护的是到1--i个数中的最小值然后,high同理。然后不断增加i, 同时j不变,直到最大-最小>=k为止,此时[j,i-1]还是可行的,那么以j为开头,末尾任选有i-j种。

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
deque<int> low,high;
int s[100009];
int main()
{
	//freopen("t.txt","r",stdin);
	int n,k,T,j;
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d%d",&n,&k);
		while(!high.empty()) high.pop_back();
		while(!low.empty()) low.pop_back();
		for(int i=0;i<n;i++)
		{
			scanf("%d",&s[i]);
		}
		int d=0;
		long long res=0;
		j=0;
		for(int i=0;i<n;i++)
		{
			while(!high.empty()&&s[i]>high.back()) high.pop_back();
			high.push_back(s[i]);
			while(!low.empty()&&s[i]<low.back()) low.pop_back();
			low.push_back(s[i]);
			while(!high.empty()&&!low.empty()&&high.front()-low.front()>=k)
			{
				res+=(i-j);
				if(high.front()==s[j]) high.pop_front();
				if(low.front()==s[j]) low.pop_front();
				j++;
			}
		}
		while(j<n)
		{
			res+=(n-j);
			j++;
		}
		printf("%lld\n",res);
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_39861441/article/details/87906838