【问题描述】
A A 君有一个长度为 的排列 P P , 现在 君想知道,有多少个连续整数组成的区间 [l,r] [ l , r ] 可以由 P P 中的两个区间 拼出,其中 1≤a≤b<c≤d≤n 1 ≤ a ≤ b < c ≤ d ≤ n 。
例如 P=[1,5,2,4,6,3] P = [ 1 , 5 , 2 , 4 , 6 , 3 ] , 那么数字区间 [5,6] [ 5 , 6 ] 可以由 P P 的区间 与 [5,5] [ 5 , 5 ] 拼出。换句话说 P P 中这两个区间数字的并集为 。
【数据范围】
10% 10 % 的数据: n≤20 n ≤ 20
40% 40 % 的数据: n≤2000 n ≤ 2000
另有 10% 10 % 的数据: Pi=i P i = i
100% 100 % 的数据: 1≤n≤3×105 1 ≤ n ≤ 3 × 10 5
【题解】
首先我们考虑部分分的做法
一开始,我们会想到枚举 a≤b≤c≤d a ≤ b ≤ c ≤ d ,并且看这两个区间的并是否是连续的数,期望得分: 10 10 。
然后,可以观察到题目问的是有多少区间 [l,r] [ l , r ] 满足题意,而不是求 a,b,c,d a , b , c , d ,因此我们考虑枚举区间 [l,r] [ l , r ] ,然后判断其在 P P 中是否为两个(或一个)区间,这样的复杂度是 的,也只有 10 10 分。但是由于我们枚举的时候,其实是在确定 l l 的情况下,将 不断从右侧加入一个数,因此我们可以维护一个 bool bool 数组表示 [l,r] [ l , r ] 中的数字在 P P 里出现在那些位置。然后每加入一个数时,判断其两侧的数,如果两侧的数都不存在则段数 ,都存在段数 −1 − 1 ,否则段数不变。最后统计段数小于三的区间 [l,r] 即可。复杂度 O(n2) O ( n 2 ) 。
可是正解要能跑 n≤100000 n ≤ 100000 啊,所以我们可能需要一个 O(nlog(n)) O ( n l o g ( n ) ) 的算法。我们发现其实按照原本这样写已经不好优化了,因此我们把这个暴力改写成我们习惯的 dp dp 的形式,便于接下来的优化。
我们记 i i 在 中的位置为 pos pos , fl,r f l , r 为区间 [l,r] [ l , r ] 会被分为 P P 中几段,那么由上可知 的方程,答案则是对于所有满足 fi,j≤2 f i , j ≤ 2 的数量。然后我们可以发现假定区间的右端点 r r 保持不变并且我们已知所有的 ,那么所有的 fi,j f i , j 可以由这样的更新方法来得到:
∙ ∙ 如果 Ppos−1 P p o s − 1 与 Ppos+1 P p o s + 1 中恰有一个数大于 i i ,令其为 ,则对于所有的 i<j≤n i < j ≤ n 的 j j ,。
∙ ∙ 如果 Ppos−1 P p o s − 1 与 Ppos+1 P p o s + 1 中均小于 i i ,则对于所有的 的 j j ,。
(以上两种情况可以合并来考虑)
∙ ∙ 如果 Ppos−1 P p o s − 1 与 Ppos+1 P p o s + 1 中均大于 i i ,令其中小的那个数为 ,大的数为 y y ,则对于所有的 的 j j ,,对于所有的 y<j≤n y < j ≤ n 的 j j ,,对于所有的 x<j≤y x < j ≤ y 的 j j ,。
那么此时,如果我们把 fi f i 滚动掉会发生什么呢,是不是变成了一个一维的数组,然后你需要实现区间上的修改,并且对于每个 i i 查询数组内 的数字个数。
我们可以方便地使用线段树来维护这个数组,对于每个叶子节点我们存放 fj f j 的值,并且我们维护区间上的最小值,次小值及其个数即可。
于是你写个带区间修改的线段树就过了(
代码如下
#include <bits/stdc++.h>
#define R register
#define LL long long
#define Max(__a,__b) (__a<__b?__b:__a)
#define Min(__a,__b) (__a<__b?__a:__b)
using namespace std;
template<class TT>inline void read(R TT &x){
x=0;R bool f=false;R char c=getchar();
for(;c<48||c>57;c=getchar())f|=(c=='-');
for(;c>47&&c<58;c=getchar())
x=(x<<1)+(x<<3)+(c^48);
(f)&&(x=-x);
}
int flag=1;
int n;
int P[300010],pos[300010];
char z[300010];
struct seg_tree{
struct tree_node{
int min[2],cnt[2],tag;
inline void add(R int x){
min[0]+=x;
min[1]+=x;
tag+=x;
}
}t[1<<20];
void build(R int l,R int r,R int i){
t[i]=(tree_node){{0,1<<30},{r-l+1,0},0};
if(l==r)return;
R int mid=l+r>>1;
build(l,mid,i<<1);
build(mid+1,r,i<<1|1);
}
inline void push_down(R int i){
if(t[i].tag){
t[i<<1].add(t[i].tag);
t[i<<1|1].add(t[i].tag);
t[i].tag=0;
}
}
void upd(R int l,R int r,R int i,R int ql,R int qr,R int qx){
if(ql<=l&&r<=qr){
t[i].add(qx);
return;
}
push_down(i);
R int mid=l+r>>1;
if(qr<=mid)upd(l,mid,i<<1,ql,qr,qx);
else if(ql>mid)upd(mid+1,r,i<<1|1,ql,qr,qx);
else upd(l,mid,i<<1,ql,qr,qx),upd(mid+1,r,i<<1|1,ql,qr,qx);
for(R int p=0,ml,mr,pl=0,pr=0;p<2;++p){
ml=t[i<<1].min[pl];
mr=t[i<<1|1].min[pr];
if(ml<mr)t[i].min[p]=ml,t[i].cnt[p]=t[i<<1].cnt[pl++];
else if(ml>mr)t[i].min[p]=mr,t[i].cnt[p]=t[i<<1|1].cnt[pr++];
else t[i].min[p]=ml,t[i].cnt[p]=t[i<<1].cnt[pl++]+t[i<<1|1].cnt[pr++];
}
}
int query(R int l,R int r,R int i,R int ql,R int qr){
if(ql<=l&&r<=qr){return (t[i].min[0]<3?t[i].cnt[0]:0)+(t[i].min[1]<3?t[i].cnt[1]:0);}
push_down(i);
R int mid=l+r>>1;
if(qr<=mid)return query(l,mid,i<<1,ql,qr);
if(ql>mid)return query(mid+1,r,i<<1|1,ql,qr);
return query(l,mid,i<<1,ql,qr)+query(mid+1,r,i<<1|1,ql,qr);
}
}T;
int main(){
read(n);
for(R int i=1;i<=n;++i){
read(P[i]);
pos[P[i]]=i;
if(P[i]!=i)flag=false;
}
if(flag)return (printf("%lld\n",(LL)n*(n-1)/2),0);
R LL ans=0;
T.build(1,n,1);
for(R int r=1;r<=n;++r){
R int t[2]={0},cnt=0;
if(pos[r]>1&&P[pos[r]-1]<r)t[cnt++]=P[pos[r]-1];
if(pos[r]<n&&P[pos[r]+1]<r)t[cnt++]=P[pos[r]+1];
if(cnt<2)T.upd(1,n,1,t[0]+1,r,1);
else{
if(t[0]>t[1])swap(t[0],t[1]);
T.upd(1,n,1,1,t[0],-1);
T.upd(1,n,1,t[1]+1,r,1);
}
ans+=T.query(1,n,1,1,r);
}
printf("%lld\n",ans-n);
return 0;
}