玲珑 1149 - Buildings 【rmq+二分】

点击打开链接

题意：

      给你一个序列，让你求出符合（区间最大值-区间最小值）<=k的的区间有多少个。

      序列大小200000

理解：

 

     很显然，n^2 会挂，

     那么用st表，或是线段树，什么的维护个区间最值，就行了， 

     查询所有区间肯定不能一个一个查，因为区间数也是n^2.因为区间最值差是单调非减的，所以二分以下区间最值差就可以了。

     O(nlogn)

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2e5+5;
const int mod=1e9+7;
#define ll long long
inline int read(){int x=0,f=1;char c=getchar();for(;!isdigit(c);c=getchar())
if(c=='-')f=-1;for(;isdigit(c);c=getchar())x=x*10+c-'0';return x*f;}
int dp[maxn][20];
int dd[maxn][20];
int mm[maxn];
void init(int n,int b[]){
    mm[0]=-1;
    for(int i=1;i<=n;++i){
        mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
        dp[i][0]=b[i];
        dd[i][0]=b[i];
    }
    for(int j=1;j<=mm[n];++j)
        for(int i=1;i+(1<<j)-1<=n;++i){
            dp[i][j]=max(dp[i][j-1],dp[i+(1<<(j-1))][j-1]);
            dd[i][j]=min(dd[i][j-1],dd[i+(1<<(j-1))][j-1]);
        }

}
int rmq1(int x,int y){
    int k=mm[y-x+1];
    return max(dp[x][k],dp[y-(1<<k)+1][k]);
}
int rmq2(int x,int y){
    int k=mm[y-x+1];
    return min(dd[x][k],dd[y-(1<<k)+1][k]);
}
int a[maxn];
int n,k;
bool judge(int L,int R){
    int mx=rmq1(L,R);
    int mn=rmq2(L,R);
    if(mx-mn<=k) return true;
    return false;
}
int main(){

    n=read();k=read();
    for(int i=1;i<=n;++i) a[i]=read();
    init(n,a);
    ll ans=0;
    int t1,t2;
    for(int i=1;i<=n;++i){
        int l=i,r=n,m;
        while(l<=r){
            m=(l+r)/2;
            if(judge(i,m)) l=m+1;
            else r=m-1;
        }
        ans+=l-i;
    }
    printf("%lld\n",ans);
    return 0;
}