rt,我是用的分治做法,每次找到对应区间里最大的数钦定它为 maxpos,然后将区间分为左右两块统计答案,单次时间复杂度是左右两个区间中较小区间的长度。
代码中线段树是找到区间中的 maxpos,树状数组用于求出每个位置的数左边第一个比他小和右边第一个比它小的数的位置。
但是我的代码在第 24 个点 MLE 了,数据是 n=106 然后序列单调递增。
// Problem: E. Max to the Right of Min
// Contest: Codeforces - Educational Codeforces Round 152 (Rated for Div. 2)
// URL: https://codeforces.com/contest/1849/problem/E
// Memory Limit: 256 MB
// Time Limit: 3000 ms
//
// Powered by CP Editor (https://cpeditor.org)
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#define ll long long
#define N 1000005
using namespace std;
int tot,a[N],le[N],ri[N];
struct Node{
int l,r,ls,rs,maxn,v;
#define l(x) tr[x].l
#define r(x) tr[x].r
#define m(x) tr[x].maxn
#define ls(x) tr[x].ls
#define rs(x) tr[x].rs
#define v(x) tr[x].v
}tr[N<<3];
void build(int &x,int l,int r){
x=++tot;l(x)=l;r(x)=r;if(l==r){v(x)=l;return;}
int mid=(l+r)>>1;
build(ls(x),l,mid);build(rs(x),mid+1,r);
if(a[v(ls(x))]>a[v(rs(x))]) v(x)=v(ls(x));
else v(x)=v(rs(x));
}
int query(int x,int l,int r){
if(l(x)>=l && r(x)<=r) return v(x);
int ans=0,mid=((l(x)+r(x))>>1);
if(l<=mid)ans=query(ls(x),l,r);
if(r>mid){
int tp=query(rs(x),l,r);
if(a[tp]>a[ans]) ans=tp;
}
return ans;
}
int b[N];
int lowbit(int x){
return x&(-x);
}
void add(int x,int val,int opt){
while(x<N){
if(opt)b[x]=max(b[x],val);
else b[x]=min(b[x],val);
x+=lowbit(x);
}
}
int ask(int x,int opt){
int ans;
if(!opt)ans=0x3f3f3f3f;
else ans=0;
while(x){
if(opt)ans=max(ans,b[x]);
else ans=min(ans,b[x]);
x-=lowbit(x);
}
return ans;
}
ll solve(int l,int r){
if(l==r || l>r) return 0;
int maxp=query(1,l,r);ll ans=0;
int tmp=le[maxp];
while(tmp>=l){
int tmp2=ri[tmp];
if(tmp2!=0x3f3f3f3f){
ll x=(tmp-max(l-1,le[tmp2])),y=(min(r+1,tmp2)-maxp);
ans+=x*y;
}
else{
ll x=(tmp-l+1),y=(r-maxp+1);
ans+=x*y;
break;
}
tmp=le[tmp2];
}
return ans+solve(l,maxp-1)+solve(maxp+1,r);
}
int main(){
int n,rt;cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
build(rt,1,n);
for(int i=1;i<=n;i++){
le[i]=ask(a[i]-1,1);
add(a[i],i,1);
}
for(int i=0;i<N;i++)b[i]=0x3f3f3f3f;
for(int i=n;i>=1;i--){
ri[i]=ask(a[i]-1,0);
add(a[i],i,0);
}
cout<<solve(1,n);
}