51Nod 1275 - 连续子段的差异(单调队列)

【题目描述】
在这里插入图片描述

【思路】
固定左端点 i i ,向右寻找一个最远的右端点 j j ,使得区间 a [ i , j ] a[i,j] 中的最大值减去最小值的差 < = k <=k 同时 a [ i , j + 1 ] a[i,j+1] 中的最大值减去最小值的差 > k >k ,这样一来 [ i , i ] , [ i , i + 1 ] . . . [ i , j ] [i,i],[i,i+1]...[i,j] 都符合要求,以 i i 为左端点的区间对答案的贡献为 j i + 1 j-i+1 ,因为随着 i i 的增加, j j 也只能不断增加,所以可以用单调队列求区间的最大值和最小值

#include<bits/stdc++.h>
using namespace std;

const int maxn=50005;

int n,k;
int a[maxn];
deque<int> minq,maxq;

int main(){
	scanf("%d%d",&n,&k);
	for(int i=0;i<n;++i) scanf("%d",&a[i]);
	long long ans=0;
	int ri=0;
	for(int i=0;i<n;++i){
		while(minq.size() && minq.front()<i) minq.pop_front();
		while(maxq.size() && maxq.front()<i) maxq.pop_front();
		while(ri<n){
			while(minq.size() && a[minq.back()]>=a[ri]) minq.pop_back();
			minq.push_back(ri);
			while(maxq.size() && a[maxq.back()]<=a[ri]) maxq.pop_back();
			maxq.push_back(ri);
			if(a[maxq.front()]-a[minq.front()]>k) break;
			++ri;
		}
		ans+=ri-i;
	}
	printf("%lld\n",ans);
	return 0;
}

猜你喜欢

转载自blog.csdn.net/xiao_k666/article/details/83794750