18 宁夏邀请赛 Continuous Intervals // 线段树 +单调栈

 题意：给数组，求有多少个[l,r]，满足元素种类数量>最大值-最小值

思路：

线段树维护max-min-cnt，cnt为元素种类数量。

线段树维护的是后缀，不断加入a[i]作为r，统计满足的l的数量。

用单调栈，考虑加入a[i]后，对最大值，最小值的影响，在单调栈pop()时更新。

/*   (o O o)
 *   Author : Rshs
 *   Data : 2019-08-31-13.50
 */
#include<bits/stdc++.h>
using namespace std;
#define FI first
#define SE second
#define LL long long
#define MP make_pair
#define PII pair<int,int>
#define SZ(a) (int)a.size()
const LL mod = 1e9+7;
const int MX = 1e5+5;
int a[MX];int n;
int lto[MX];
map<int,int>ma,pos;
stack<int>su,sd;
int lz[MX<<2],mi[MX<<2],cnt[MX<<2];
void pushUp(int rt){
    mi[rt]=min(mi[rt<<1],mi[rt<<1|1]);
    if(mi[rt<<1]==mi[rt<<1|1]) cnt[rt]=cnt[rt<<1]+cnt[rt<<1|1];
    if(mi[rt<<1]>mi[rt<<1|1]) cnt[rt]=cnt[rt<<1|1];
    if(mi[rt<<1]<mi[rt<<1|1]) cnt[rt]=cnt[rt<<1];
}
void pushDown(int rt){
    if(lz[rt]){
        lz[rt<<1]+=lz[rt];
        lz[rt<<1|1]+=lz[rt];
        mi[rt<<1]+=lz[rt];
        mi[rt<<1|1]+=lz[rt];
        lz[rt]=0;
    }
}
void update(int L,int R,int c,int l,int r,int rt){
    if(l>=L&&r<=R){
        lz[rt]+=c;
        mi[rt]+=c;
        return;
    }
    pushDown(rt);
    int m=(l+r)>>1;
    if(m>=L) update(L,R,c,l,m,rt<<1);
    if(m<R)  update(L,R,c,m+1,r,rt<<1|1);
    pushUp(rt);
}
void UU(int l,int r,int rt,int pp){
    if(l==r){
        mi[rt]=-1;cnt[rt]=1;
        return ;
    }
    pushDown(rt);
    int mid=(l+r)>>1;
    if(pp<=mid) UU(l,mid,rt<<1,pp);
    else UU(mid+1,r,rt<<1|1,pp);
    pushUp(rt);
}

void Main(int avg){
    cin>>n;
    for(int i=1;i<=4*n;i++) mi[i]=INT_MAX,lz[i]=0,cnt[i]=0;
    while(!sd.empty())sd.pop();
    while(!su.empty())su.pop();
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    for(int i=1;i<=n;i++) lto[i]=1;ma.clear(),pos.clear();
    for(int i=1;i<=n;i++) {
        if(ma[a[i]]>0) lto[i]=pos[a[i]]+1;
        ma[a[i]]++;pos[a[i]]=i;
    }
    LL ans=0;
    for(int i=1;i<=n;i++){
        UU(1,n,1,i);
        while(!su.empty()&&a[i]<=a[su.top()]){
            int gg=su.top();su.pop();
            int now=1;if(!su.empty())now=su.top()+1;
            update(now,gg,-a[i]+a[gg],1,n,1);
        }
        su.push(i);
        while(!sd.empty()&&a[i]>=a[sd.top()]){
            int gg=sd.top();sd.pop();
            int now=1;if(!sd.empty())now=sd.top()+1;
            update(now,gg,a[i]-a[gg],1,n,1);
        }
        sd.push(i);
        if(lto[i]<=i-1)update(lto[i],i-1,-1,1,n,1);
        ans=ans+(LL)cnt[1];
    }
    printf("Case #%d: ",avg);cout<<ans<<'\n';
}
int main(){
    int ca;cin>>ca;for(int i=1;i<=ca;i++)Main(i);
    return 0;
}
/*
10
4
1 3 2 4
*/