【牛客网】NC13134 牛牛的数列
题号:NC13134
时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 32768K,其他语言65536K
64bit IO Format: %lld
题目描述
牛牛现在有一个n个数组成的数列,牛牛现在想取一个连续的子序列,并且这个子序列还必须得满足:最多只改变一个数,就可以使得这个连续的子序列是一个严格上升的子序列,牛牛想知道这个连续子序列最长的长度是多少。
输入描述
输入包括两行,第一行包括一个整数n(1 ≤ n ≤ 10^5),即数列的长度;
第二行n个整数a_i, 表示数列中的每个数(1 ≤ a_i ≤ 10^9),以空格分割。
输出描述
输出一个整数,表示最长的长度。
示例
输入
6
7 2 3 1 5 6
输出
5
思路
- 用动态规划的思想,利用两个数组front、rear,front用来存储以当前字符作为结尾的最长严格递增子序列,rear用来存储以当前字符作为起始字符的最长严格递增子序列。
- 符合题目要求的最长子序列就是以每个字符为断点,其前一个字符作为结尾的最长严格递增子序列和其后一个字符作为起始字符的最长严格递增子序列之和再加一的最大值。即front[i-1]+rear[i+1]+1。其他判断条件见代码。
- 例如给出的例子中:以1为断点,其前一个字符3作为结尾的严格递增子序列为2、3,长度为2;其后一个字符5作为起始的严格递增子序列为5、6,长度为2,所以符合题目的最长子序列的长度为2+2+1=5.
代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
const int maxn=1e5+10;
int front[maxn];
int rear[maxn];
int arr[maxn];
int main(){
int n,answer=0;
cin>>n;
for(int i=0;i<n;i++){
cin>>arr[i];
}
//以当前数字作为结尾的最长子序列
front[0]=1;
for(int i=1;i<n;i++){
if(arr[i]>arr[i-1])
front[i]=front[i-1]+1;
else
front[i]=1;
}
//以当前数字作为开始的最长子序列
//用从后往前的方法查询
rear[n-1]=1;
for(int i=n-2;i>=0;i--){
if(arr[i]<arr[i+1])
rear[i]=rear[i+1]+1;
else
rear[i]=1;
}
//以每个数为断点考察其前后的最长子序列
for(int i=1;i<n-1;i++) {
if(arr[i+1]-arr[i-1]>1) //当前数的前后两个数之差大于1
answer=max(answer,front[i-1]+rear[i+1]+1);
else{
//如果当前数字的前后两个数字相差<=1,例1、2、3、3、4、5
//则等于其前后最长子序列中最长的那一个+1
answer=max(answer,front[i-1]+1);
answer=max(answer,rear[i+1]+1);
}
}
cout<<answer;
}
总结
最初想练练动态规划的题,没想到这道题最难的不是动态规划的部分,反反复复纠结了很久,以上代码已通过,欢迎指正。