【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\n",ans);

}

int main(){

	int T;scanf("%d",&T);

	while(T--){

		read_and_parse();

		solve();

	}

	return 0;

}

巴特西

【HDU5289】Assignment

最新文章

热门文章