题目大意:给定一个长度为 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;
}