【枚举】Mushroom的序列

【问题描述】
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;
}