【Codeforces193D】Two Segments

枚举线段树lazy标记维护即可。（似乎有点太久远了）

#include <bits/stdc++.h>
#define N 300009
#define gc getchar()
#define ll long long
#define mid (l+r>>1)
using namespace std;
int n,a[N],pos[N];
ll ans;
int dl[N],dr[N],now[N],cnt,root;
struct node
{
    int ls,rs,sum,min,len,bz;
}seg[5000000];
int read()
{
    int x=1;
    char ch;
    while (ch=gc,ch<'0'||ch>'9') if (ch=='-') x=-1;
    int s=ch-48;
    while (ch=gc,ch>='0'&&ch<='9') s=s*10+ch-48;
    return s*x;
}
void build(int l,int r,int &cur)
{
    if (!cur) cur=++cnt;
    if (l==r) seg[cur].len=1;
    if (l==r) return;
    build(l,mid,seg[cur].ls);
    build(mid+1,r,seg[cur].rs);
}
void down(int cur)
{
    if (seg[cur].bz)
    {
        seg[seg[cur].ls].min+=seg[cur].bz;
        seg[seg[cur].ls].bz+=seg[cur].bz;
        seg[seg[cur].rs].min+=seg[cur].bz;
        seg[seg[cur].rs].bz+=seg[cur].bz;
        seg[cur].bz=0;
    }
}
void up(int cur)
{
    seg[cur].min=min(seg[seg[cur].ls].min,seg[seg[cur].rs].min);
    seg[cur].len=seg[seg[cur].ls].len*(seg[cur].min==seg[seg[cur].ls].min);
    seg[cur].len+=seg[seg[cur].rs].len*(seg[cur].min==seg[seg[cur].rs].min);
    seg[cur].sum=seg[seg[cur].ls].sum*(seg[cur].min==seg[seg[cur].ls].min);
    seg[cur].sum+=seg[seg[cur].rs].sum*(seg[cur].min==seg[seg[cur].rs].min);
    seg[cur].sum+=seg[seg[cur].ls].len*(seg[cur].min+1==seg[seg[cur].ls].min);
    seg[cur].sum+=seg[seg[cur].rs].len*(seg[cur].min+1==seg[seg[cur].rs].min);
}
void ins(int l,int r,int R,int k,int cur)
{
    if (r<=R)
    {
        seg[cur].min+=k;
        seg[cur].bz+=k;
        return;
    }
    down(cur);
    ins(l,mid,R,k,seg[cur].ls);
    if (mid<R) ins(mid+1,r,R,k,seg[cur].rs);
    up(cur);
}
int qry(int l,int r,int R,int cur)
{
    if (r<=R) return seg[cur].sum*(seg[cur].min<2)+seg[cur].len*(seg[cur].min<3);
    down(cur);
    int ans=qry(l,mid,R,seg[cur].ls);
    if (mid<R) ans+=qry(mid+1,r,R,seg[cur].rs);
    return ans;
}
int main()
{
    n=read();
    build(1,n,root);
    for (int i=1;i<=n;i++)
        a[i]=read(),pos[a[i]]=i;
    for (int i=1;i<=n;i++)
    {
        ins(1,n,i,1,root);
        if (pos[i]-1>0&&a[pos[i]-1]<i) ins(1,n,a[pos[i]-1],-1,root);
        if (pos[i]+1<=n&&a[pos[i]+1]<i) ins(1,n,a[pos[i]+1],-1,root);
        if (i>1) ans+=qry(1,n,i-1,root);
    }
    printf("%lld\n",ans);
    return 0;
}