【HDU5289】Assignment

题目大意:给定一个长度为 N 的序列,求序列中最大值和最小值相差小于 K 的连续段的个数。

题解:
最大值和最小值相差不超过 K 是一个在值域角度的限制,应考虑采用平衡树或权值...数据结构进行维护。
连续的一段即可考虑采用双指针进行维护即可。

注意:

  • r 是当前需要判断的情况,故 r 从 1 开始。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
typedef long long LL;

int n,k,a[maxn];
multiset<int> s;

void read_and_parse(){
	s.clear();
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)scanf("%d",&a[i]);
}
bool right(int val){
	if(s.empty())return 1;
	else return abs(*s.begin()-val)<k&&abs(*--s.end()-val)<k;
}
void solve(){
	LL ans=0;
	for(int l=1,r=1;l<=n;l++){
		while(r<=n&&right(a[r]))s.insert(a[r++]);
		ans+=r-l;
		s.erase(s.find(a[l]));
	}
	printf("%lld
",ans);
}
int main(){
	int T;scanf("%d",&T);
	while(T--){
		read_and_parse();
		solve();
	}
	return 0;
}
原文地址:https://www.cnblogs.com/wzj-xhjbk/p/10929180.html