st表应用

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

题目描述

给出长度为nnn的序列a1,a2,...,ana_1,a_2,...,a_na1​,a2​,...,an​及一个数kkk,定义一段区间[l,r][l,r][l,r]的美丽值为max(a[i]..a[j])=min(a[i]..a[j])+k(l≤i≤j≤r)max(a[i]..a[j])=min(a[i]..a[j])+k(l\le i\le j \le r)max(a[i]..a[j])=min(a[i]..a[j])+k(l≤i≤j≤r)的个数
求区间[1,n][1,n][1,n]的美丽值

输入描述:

第一行两个正整数n,k{n,k}n,k

第二行nnn个整数代表aia_iai​

1≤n,ai≤106,0≤k≤106{1\le n,a_i\le 10^6, 0\le k\le 10^6}1≤n,ai​≤106,0≤k≤106

输出描述:

输出一行一个数代表答案

#include<iostream>
using namespace std;
const int maxn=1e6+5;
typedef long long ll;
int lg[maxn];
int a[maxn][20];
int b[maxn][20];
int stmx(int l,int r)
{
    int t=lg[r-l+1];
    return max(a[l][t],a[r-(1<<t)+1][t]);
}
int stmn(int l,int r)
{
    int t=lg[r-l+1];
    return min(b[l][t],b[r-(1<<t)+1][t]);
}
int main()
{
    lg[1]=0;
    for(int i=2;i<maxn;i++)
    {
        lg[i]=lg[i/2]+1;
    }
    int n,k;cin>>n>>k;
    for(int i=1;i<=n;i++){
        cin>>a[i][0];
        b[i][0]=a[i][0];
    }
    for(int i=1;(1<<i)<=n;i++)
    {
        for(int j=1;j+(1<<(i-1))<=n;j++)
        {
            a[j][i]=max(a[j+(1<<(i-1))][i-1],a[j][i-1]);
            b[j][i]=min(b[j+(1<<(i-1))][i-1],b[j][i-1]);
        }
    }
    ll ans=0;
    for(int i=1;i<=n;i++)
    {
        int l=i,r=n;
        int pos1=n+1;
        while(l<=r)
        {
            int m=(l+r)/2;
            int mx=stmx(i,m),mn=stmn(i,m);
            if(mx-mn>=k)
            {
                pos1=m;
                r=m-1;
            }
            else l=m+1;
        }
        l=i,r=n;
        int pos2=n+1;
        while(l<=r)
        {
            int m=(l+r)/2;
            int mx==stmx(i,m),mn=stmn(i,m);
            if(mx-mn>k)
            {
                pos2=m;
                r=m-1;
            }
            else l=m+1;
        }
        ans+=pos2-pos1;
    }
    cout<<ans<<endl;
}