10.21T5 二分+线段树

Description

Input

6
10
3
7
4
12
2

Output

5

 

 

 

对每个位置二分后面的位置，找出恰好没有高于自己的点加起来就可以了，用线段树维护最大值

code：

#include<iostream>
#include<cstdio>
#define N 1000005
#define lc (p<<1)
#define rc (p<<1|1)
using namespace std;
long long a[N];
struct node{
    long long l,r,max;
}t[N];
void pushup(long long p){
    t[p].max=max(t[lc].max,t[rc].max);
}
void build(long long p,long long l,long long r){
    t[p].l=l,t[p].r=r;
    if(l==r){
        t[p].max=a[l];
        return;
    }
    long long mid=l+r>>1;
    build(lc,l,mid);
    build(rc,mid+1,r);
    pushup(p);
}
long long query(long long p,long long ql,long long qr){
    if(ql<=t[p].l&&t[p].r<=qr){
        return t[p].max;
    }
    long long ans=0;
    long long mid=t[p].l+t[p].r>>1;
    if(ql<=mid)ans=max(ans,query(lc,ql,qr));
    if(qr>mid)ans=max(ans,query(rc,ql,qr));
    return ans;
}
int main(){
    long long n;
    cin>>n;
    for(long long i=1;i<=n;i++){
        cin>>a[i];
    }
    build(1,1,n);
    long long tot=0;
    for(long long i=1;i<=n;i++){
        long long l=i,r=n,ans=0;
        while(l<=r){
            long long mid=l+r>>1;
            if(query(1,i+1,mid)<a[i])l=mid+1,ans=mid;
            else r=mid-1;
        }
        tot+=ans-i;
    }
    cout<<tot;
}

over