题目:
给定一个长度为
n
n
n的序列
a
a
a,若序列
b
b
b满足
min
(
b
1
,
b
2
)
≤
min
(
b
2
,
b
3
)
≤
.
.
.
≤
min
(
b
k
−
1
,
b
k
)
\min(b_1,b_2) \le \min(b_2,b_3) \le ... \le \min(b_{k-1},b_k)
min(b1,b2)≤min(b2,b3)≤...≤min(bk−1,bk),其中
k
k
k为序列
b
b
b的长度,那么称序列
b
b
b为几乎上升的,求序列
a
a
a的最长的几乎上升的子序列的长度。
(
1
≤
n
≤
5
×
1
0
5
,
1
≤
a
i
≤
n
)
(1 \le n \le 5 \times 10^5,1 \le a_i \le n)
(1≤n≤5×105,1≤ai≤n)
题解:
首先考虑一个
d
p
dp
dp,令
d
p
i
dp_i
dpi为以
a
i
a_i
ai为结尾的最长几乎上升子序列的长度,转移的时候考虑倒数第二个数是大于
a
i
a_i
ai的还是小于
a
i
a_i
ai的分两种情况,转移方程为
d
p
i
=
{
d
p
j
+
1
j
<
i
&&
a
j
≤
a
i
d
p
j
+
2
,
j
<
i
&&
a
j
≤
a
i
&&
∃
x
∈
[
j
+
1
,
i
−
1
]
使
a
x
≥
a
i
,
a
x
≥
a
j
dp_i = \begin{cases} dp_j+1 & j <i \text{ \&\& }a_j \le a_i \\ dp_j+2, & j<i \text{ \&\& }a_j \le a_i \text{ \&\& } \exist x \in [j+1,i-1]\text{使}a_x \ge a_i,a_x \ge a_j \end{cases}
dpi={dpj+1dpj+2,j<i && aj≤aij<i && aj≤ai && ∃x∈[j+1,i−1]使ax≥ai,ax≥aj
这个
d
p
dp
dp是
O
(
n
2
)
O(n^2)
O(n2)的,需要优化。对于第一个转移我们可以用线段树维护
(
a
i
,
d
p
i
)
(a_i,dp_i)
(ai,dpi)的信息,然后找
[
1
,
a
i
−
1
]
[1,a_i-1]
[1,ai−1]中的最大值转移即可。对于第二个转移,我们可以先找到一个离
a
i
a_i
ai最近的大于等于
a
i
a_i
ai的数,这个可以用单调栈实现,假设这个数的位置为
p
p
p,那么可以从满足
a
x
≥
a
i
,
x
∈
[
1
,
p
−
1
]
a_x \ge a_i,x \in [1,p-1]
ax≥ai,x∈[1,p−1]的位置
x
x
x转移过来,这里同时涉及权值和区间信息,用主席树维护即可。
复杂度: O ( n l o g n ) O(nlogn) O(nlogn)
代码:
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<queue>
#include<stack>
#include<map>
#include<set>
#include<string>
#include<bitset>
#include<sstream>
#include<ctime>
//#include<chrono>
//#include<random>
//#include<unordered_map>
using namespace std;
#define ll long long
#define pii pair<int,int>
#define fi first
#define se second
#define pb push_back
#define mp make_pair
#define sz(x) (int)(x).size()
#define all(x) (x).begin(),(x).end()
const double pi=acos(-1.0);
const double eps=1e-6;
const int mod=1e9+7;
const int INF=0x3f3f3f3f;
const int maxn=5e5+5;
ll read(){
ll x=0,f=1;
char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int t,n;
int a[maxn],dp[maxn],stk[maxn],rt[maxn];
struct PST{
struct node{
int ls,rs,v;
}t[maxn*40];
int tot;
void init(){
tot=0;
}
void build(int &cur,int l,int r){
cur=++tot;
t[cur].v=0;
if(l==r)return;
int m=(l+r)>>1;
build(t[cur].ls,l,m);
build(t[cur].rs,m+1,r);
}
void ins(int pre,int &cur,int l,int r,int p,int v){
cur=++tot;
t[cur]=t[pre];
t[cur].v=max(t[cur].v,v);
if(l==r)return;
int m=(l+r)>>1;
if(p<=m)ins(t[pre].ls,t[cur].ls,l,m,p,v);
else ins(t[pre].rs,t[cur].rs,m+1,r,p,v);
}
int qry(int cur,int l,int r,int p){
if(r<=p){
return t[cur].v;
}
int m=(l+r)>>1;
if(p<=m){
return qry(t[cur].ls,l,m,p);
}
else{
return max(t[t[cur].ls].v,qry(t[cur].rs,m+1,r,p));
}
}
}T;
int main(void){
// freopen("in.txt","r",stdin);
scanf("%d",&t);
while(t--){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
T.init();
T.build(rt[0],1,n);
int tp=0,ans=0;
for(int i=1;i<=n;i++){
int res=T.qry(rt[i-1],1,n,a[i]);
dp[i]=res+1;
while(tp>0&&a[stk[tp]]<a[i]){
--tp;
}
if(tp){
int res=T.qry(rt[stk[tp]-1],1,n,a[i]);
dp[i]=max(dp[i],res+2);
}
stk[++tp]=i;
T.ins(rt[i-1],rt[i],1,n,a[i],dp[i]);
ans=max(ans,dp[i]);
}
printf("%d\n",ans);
}
return 0;
}