HDU 5289 思路+线段树处理

给出N个数，和K

求这N个数的所有满足条件（最大数-最小数<k)的区间个数

数组b记录以当前位置开始，到最右端最多满足条件的数的个数，b数组的值可通过二分+线段树查找区间最大最小值得到

对于第二组数据

10 5

0 3 4 5 2 1 6 7 8 9

B数组为

3 7 7 7 3 1 4 3 2 1

由于当前I点取最右端值可能会导致i+1点和后面取到的点不满足条件，所有应有：b[i]<=b[i+1]+1

得B数组

3 5 4 3 2 1 4 3 2 1

ans=28

#include "stdio.h"
#include "string.h"

struct node
{
    int l,r;
    __int64 ma,mi;
}data[400010];
__int64 a[100010],b[100010];
__int64 Max(__int64 a,__int64 b)
{
    if (a<b) return b;
    else return a;
}

__int64 Min(__int64 a,__int64 b)
{
    if (a<b) return a;
    else return b;
}

void build(int l,int r,int k)
{
    int mid;
    data[k].l=l;
    data[k].r=r;
    if (l==r)
    {
        data[k].ma=data[k].mi=a[data[k].l];
        return ;
    }

    mid=(data[k].l+data[k].r)/2;

    build(l,mid,k*2);
    build(mid+1,r,k*2+1);
    data[k].ma=Max(data[k*2].ma,data[k*2+1].ma);
    data[k].mi=Min(data[k*2].mi,data[k*2+1].mi);
}

__int64 query(int l,int r,int k,int op)
{
    int mid;
    if (data[k].l==l && data[k].r==r)
    {
        if (op==0) return data[k].mi;
        else return data[k].ma;
    }

    mid=(data[k].l+data[k].r)/2;
    if (r<=data[k*2].r)
        return query(l,r,k*2,op);
    else
        if (l>data[k*2].r)
        return query(l,r,k*2+1,op);
    else
    {
        if (op==0) return Min(query(l,mid,k*2,op),query(mid+1,r,k*2+1,op));
        else return Max(query(l,mid,k*2,op),query(mid+1,r,k*2+1,op));
    }
}
int main()
{
    int t,n,k,i,l,r,mid;
    __int64 ans,ma,mi;
    scanf("%d",&t);
    while (t--)
    {
        scanf("%d%d",&n,&k);
        for (i=1;i<=n;i++)
            scanf("%I64d",&a[i]);
        build(1,n,1);

        for (i=1;i<n;i++)
        {
            l=i;r=n;
            if (data[1].ma<a[i]+k && data[1].mi>a[i]-k) l=n+1;
            while (l<=r)
            {
                mid=(l+r)/2;
                ma=query(l,mid,1,1);
                mi=query(l,mid,1,0);
                if (ma>=a[i]+k || mi<=a[i]-k) r=mid-1;
                else l=mid+1;
            }
            b[i]=l-i;
        }

        b[n]=1;

        for (i=n-1;i>=1;i--)
            if (b[i]>b[i+1]+1)
            b[i]=b[i+1]+1;

        ans=0;
        for (i=1;i<=n;i++)
            ans+=b[i];
        printf("%I64d\n",ans);
    }
    return 0;
}