1149

来源:互联网 发布:汽车轮毂数据大全 编辑:程序博客网 时间:2024/05/16 18:14

题目链接:传送门

题目大意:有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 200009int 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;}
原创粉丝点击