【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;
}

转载于:https://www.cnblogs.com/wzj-xhjbk/p/10929180.html