1. 首页
2. RMQ, 二分
3. 本页

HDU 5289 Assignment (RMQ+二分)

题意：求存在最大差值小于给定K值的区间段个数。

思路：利用RMQ求出区间最大最小值，再枚举右端点，二分区间找到满足要求的最大区间累加~~

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef pair<int, int> PI;
typedef pair< PI, int> PII;
const double eps=1e-5;
const double pi=acos(-1.0);
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const int mx=1100;
#define lson l, m, rt<<1
#define rson m+1, r, rt<<1|1
const int MAXN=100000+10;

//int dp[MAXN][20];
int maxsum[MAXN][20];
int minsum[MAXN][20];
int mm[MAXN];
int b[MAXN];
int n;

void intRMQ(int n)
{
    mm[0]=-1;
    for(int i=1; i<=n; ++i)
    {
        mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
        //计算长度 n为2的倍数 n&n-1==0
        //dp[i][0]=b[i];  //初值
        maxsum[i][0]=b[i];
        minsum[i][0]=b[i];
    }
    for(int j=1; j<=mm[n]; ++j)  //for(int j=1;(1<<j)<=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]);
            maxsum[i][j] = max(maxsum[i][j - 1], maxsum[i + (1 << (j - 1))][j - 1]);
            minsum[i][j] = min(minsum[i][j - 1], minsum[i + (1 << (j - 1))][j - 1]);
        }

}

int rmq(int x,int y,bool sta)
{
    int k=mm[y-x+1];
    if(sta)
        return max(maxsum[x][k],maxsum[y-(1<<k)+1][k]);
    else
        return min(minsum[x][k],minsum[y-(1<<k)+1][k]);
}


int main()
{
    int t;
    scanf("%d",&t);
    int k;
    while(t--)
    {
        scanf("%d%d",&n,&k);
        for(int i=1; i<=n; ++i)
        {
            scanf("%d",&b[i]);
        }
        intRMQ(n);
        LL cnt=0;
        int st=1;
        for(int i=1; i<=n; ++i)
        {
            int l=st;
            int r=i;
            int pos=0;
            while(l<=r)
            {
                int mid=(l+r)>>1;
                int mx=rmq(mid,i,1);
                int mi=rmq(mid,i,0);
                if(mx-mi>=k)
                {
                    //pos+=mid-l;
                    l=mid+1;
                }
                else if(mx-mi<k)
                {
                    r=mid-1;
                }
            }
            st=l;
            cnt+=i-l+1;
        }
        cout<<cnt<<endl;
    }

    return 0;
}