E. Alice and the Unfair Game（推导线段树）

题：https://codeforces.com/contest/1236/problem/E

粗自：https://www.cnblogs.com/YSFAC/p/11715522.html

#include<bits/stdc++.h>
using namespace std;
#define lson root<<1,l,midd
#define rson root<<1|1,midd+1,r
typedef long long ll;
const int M=1e5+;
ll a[M],tree[M<<],lz[M<<];
ll L[M],R[M];
void build(ll root,ll l,ll r){
    if(l==r){
        tree[root]=l;
        return ;
    }
    int midd=(l+r)>>;
    build(lson);
    build(rson);
}
ll query(ll pos,ll root,ll l,ll r){
    if(l==r)
        return tree[root]+lz[root];
    ll midd=(l+r)>>;
    if(pos<=midd)
        return query(pos,lson)+lz[root];
    else
        return query(pos,rson)+lz[root];
}
void add(int L,int R,int c,int root,int l,int r){
    if(L<=l&&r<=R){
        lz[root]+=c;
        return ;
    }
    int midd=(l+r)>>;
    if(L<=midd)
        add(L,R,c,lson);
    if(R>midd)
        add(L,R,c,rson);
}
int main(){
    ll n,m;
    scanf("%I64d%I64d",&n,&m);
    for(int i=;i<=m;i++)
        scanf("%I64d",&a[i]);
    if(n==)
        return cout<<"",;
    //计算向右最多能走的距离
    build(,,n);
    for(int i=;i<=m;i++){
        ll l=,r=n;
        ll qul=,qur=;
        while(l<=r){
            ll midd=(l+r)>>;
            if(query(midd,,,n)<a[i]-i)
                qul=midd,l=midd+;
            else
                r=midd-;
        }
        qul++,l=,r=n;
        while(l<=r){
            ll midd=(l+r)>>;
            if(query(midd,,,n)<=a[i]-i)
                qur=midd,l=midd+;
            else
                r=midd-;
        }
        if(qul<=qur)
            add(qul,qur,-,,,n);
    }
    for(int i=;i<=n;i++)
        R[i]=min(query(i,,,n)+m+,n);
    memset(tree,,sizeof(tree));
    memset(lz,,sizeof(lz));
    //计算向左最多能走的距离
    build(,,n);
    for(int i=;i<=m;i++){
        ll l=,r=n;
        ll qul=,qur=;
        while(l<=r){
            ll midd=(l+r)>>;
            if(query(midd,,,n)<a[i]+i)
                qul=midd,l=midd+;
            else
                r=midd-;
        }
        qul++,l=,r=n;
        while(l<=r){
            ll midd=(l+r)>>;
            if(query(midd,,,n)<=a[i]+i)
                qur=midd,l=midd+;
            else
                r=midd-;
        }
        if(qul<=qur)
            add(qul,qur,,,,n);
    }
    for(int i=;i<=n;i++)
        L[i]=max(query(i,,,n)-m-,1ll);
    ll ans=;
    for(int i=;i<=n;i++)
        ans+=R[i]-L[i]+;
    printf("%I64d",ans);
}