HDU 5289 Assignment ST + 二分区间

最新推荐文章于 2023-09-11 22:19:50 发布

SolarDomo

最新推荐文章于 2023-09-11 22:19:50 发布

阅读量313

点赞数

分类专栏： ----------- 算法 ---------- 数据结构

本文链接：https://blog.csdn.net/SolarDomo/article/details/52542199

版权

----------- 算法 ---------- 同时被 2 个专栏收录

138 篇文章 0 订阅

订阅专栏

数据结构

41 篇文章 0 订阅

订阅专栏

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5289
题意：求多少个区间[l,r] 有MAX[l,r] - MIN[l,r] < k
令F[l,r] = MAX[l,r] - MIN[l,r]
对于固定的左区间l，F[l,r]是一个非减序列，所以我们可以二分找到小于K的最右边界
ST预处理一边就可以在O（1）计算F值
代码如下：

#include <bits/stdc++.h>

using namespace std;
const int maxn = 100000 + 5;
int a[maxn];
int max_st[maxn][32],min_st[maxn][32];
void init_st(int n){
    for(int i = 1;i <= n;++i) max_st[i][0] = min_st[i][0] = a[i];
    for(int j = 1;1 << j <= n;++j){
        for(int i = 1;i + (1 << j) - 1 <= n;++i){
            max_st[i][j] = max(max_st[i][j - 1],max_st[i + (1 << (j - 1))][j - 1]);
            min_st[i][j] = min(min_st[i][j - 1],min_st[i + (1 << (j - 1))][j - 1]);
        }
    }
}

int min_rmq(int l,int r){
    int k = 0;
    while(l + (1 << (k + 1)) - 1 <= r) k++;
    return min(min_st[l][k],min_st[r - (1 << k) + 1][k]);
}

int max_rmq(int l,int r){
    int k = 0;
    while(l + (1 << (k + 1)) - 1 <= r) k++;
    return max(max_st[l][k],max_st[r - (1 << k) + 1][k]);
}
int query(int l,int r){
    return max_rmq(l,r) - min_rmq(l,r);
}
typedef __int64 INT;
int main(){
    int T;scanf("%d",&T);
    while( T-- ){
        int n,k;
        INT ans = 0;scanf("%d %d",&n,&k);
        for(int i = 1;i <= n;++i){
            scanf("%d",&a[i]);
        }
        init_st(n);
        for(int i = 1;i <= n;++i){
            int l = i , r = n,mid;
            while(l < r){
                mid = (l + r + 1) >> 1;
                if(query(i,mid) < k) l = mid;
                else r = mid - 1;
            }
            ans += (l - i + 1);
        }
        printf("%I64d\n",ans);
    }
}