1149 - Buildings(RMQ+二分)

题目链接：传送门

题目大意：有n,k,有一个数组含n个数，数组有多少个子区间，区间内最大值和最小值的差小于等于k。

思路：用RMQ求出，个区间的最大值和最小值，再用二分，求满足要求的区间个数。

RMQ的详解：http://blog.csdn.net/nhl19961226/article/details/76340404

#include<bits/stdc++.h>
#include<math.h>
#include<algorithm>
using namespace std;
#define N 200009

int dp_max[N][20];//最大值
int dp_min[N][20];//最小值
int a[N];

void init(int n)//RMQ
{
    memset(dp_max,0,sizeof(dp_max));
    memset(dp_min,127,sizeof(dp_min));
    for(int i=1; i<=n; i++)
        dp_max[i][0]=dp_min[i][0]=a[i];
    for(int i=1; (1<<i)<=n; i++)
    {
        for(int j=1; j+(1<<i)-1<=n; j++)
        {
            dp_max[j][i]=max(dp_max[j][i-1],dp_max[j+(1<<(i-1))][i-1]);
            dp_min[j][i]=min(dp_min[j][i-1],dp_min[j+(1<<(i-1))][i-1]);
        }
    }
}
bool pan(int s,int e,int m)
{
    int k=log2(e-s+1);
    int max_=max(dp_max[s][k],dp_max[e-(1<<k)+1][k]);
    int min_=min(dp_min[s][k],dp_min[e-(1<<k)+1][k]);
    if(max_-min_<=m)
        return 1;
    return 0;
}
int main()
{
    int n,m;
    scanf("%d%d",&n,&m);
    for(int i=1; i<=n; i++)
        scanf("%d",&a[i]);
    init(n);//RMQ求最值
    long long sum=0;
    for(int i=1; i<=n; i++)
    {
        int l=i,r=n,mid,k=i-1;
        while(l<=r)//二分查找
        {
            int mid=(l+r)>>1;
            if(pan(i,mid,m))
            {
                l=mid+1;
                k=mid;
            }
            else r=mid-1;
        }
        sum+=(k+1-i);
    }
    cout<<sum<<endl;
    return 0;
}