【问题描述】
Mushroom手中有n个数排成一排,现在Mushroom想取一个连续的子序列,使得这个子序列满足:最多只改变一个数,使得这个连续的子序列是严格上升子序列,Mushroom想知道这个序列的最长长度是多少。
【输入格式】
第一行一个整数n,表示有n个数。
第二行为n个数。
【输出格式】
一个数,为最长长度。
【输入样例】
6
7 2 3 1 5 6
【输出样例】
5
【样例解释】
选择第2个数到第6个数,把1改变成4即可。
【数据范围】
对于30%的数据,n<=10
对于60%的数据,n<=1000
对于100%的数据,n<=100000
Solutions
考虑对于任意一个点进行修改。
对于序列 ai-1 ai ai+1
有以下三种情况
如果 ai<ai-1 最优的改法是让 ai=(ai-1)+1
如果 ai>ai+1 最优的改法是让 ai=(ai+1)-1
如果 ai>ai-1 且 ai<ai+1 说明 ai已经是最优 修改 ai不能对答案,即最长满足条件的序列做出贡献。
按照上诉方法修改之后 序列就变成 ai-1 ai' ai+1
很明显ai-1之前的最长上升序列是可以提前预处理出来的。记为pre[i]
同理 ai+1之后的最长上升序列也是可以提前预处理出来的。记为suf[i]
考虑修改的ai’对于答案的贡献
按照第一种方法修改
if(a[i]<a[i+1]) ans=max(ans,pre[i]+suf[i]-1)
if(a[i]>a[i+1]) ans=max(ans,pre[i])
同理 按照第二种方法修改
if(a[i]>a[i-1]) ans=max(ans,pre[i]+suf[i]-1)
if(a[i]<a[i-1]) ans=max(ans,suf[i])
注意的是修改之后一定要再改回去。
以下是代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
#include<vector>
using namespace std;
int n,a[100005],ans;
int pre[100005],suf[100005];
int main()
{
//freopen("seq.in","r",stdin);
scanf("%d",&n);
for (int i=1;i<=n;i++)
scanf("%d",&a[i]);
for (int i=1;i<=n;i++){
pre[i]=1;
if(a[i]>a[i-1])
pre[i]=pre[i-1]+1;
}
for (int i=n;i>=1;i--){
suf[i]=1;
if(a[i]<a[i+1])
suf[i]=suf[i+1]+1;
}
for (int i=1;i<=n;i++){
ans=max(ans,pre[i]+suf[i]-1);
if(a[i]<=a[i-1]){
int rec1=a[i],rec2=pre[i],w=0;
pre[i]=pre[i-1]+1;
a[i]=a[i-1]+1;
if(a[i]>=a[i+1]) w=-1;
if(w==0) ans=max(ans,pre[i]+suf[i]-1);
if(w==-1) ans=max(ans,pre[i]);
a[i]=rec1;pre[i]=rec2;
}
if(a[i]>=a[i+1]){
int rec1=a[i],rec2=suf[i],w=0;
suf[i]=suf[i+1]+1;
a[i]=a[i+1]-1;
if(a[i]<=a[i-1]) w=-1;
if(w==0) ans=max(ans,suf[i]+pre[i]-1);
if(w==-1) ans=max(ans,suf[i]);
a[i]=rec1;suf[i]=rec2;
}
}
printf("%d",ans);
return 0;
}