牛客编程巅峰赛S1第2场 - 青铜&白银B - 疯狂过山车(暴力)
链接:https://ac.nowcoder.com/acm/contest/6219/B
来源:牛客网
时间限制:C/C++ 5秒,其他语言10秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld
题目描述
今天牛牛去游乐园玩过山车项目,他觉得过山车在上坡下坡的过程是非常刺激的,回到家之后就受到启发,想到了一个问题。如果把整个过山车的轨道当作是一个长度为n的数组num,那么在过山车上坡时数组中的值是呈现递增趋势的,到了最高点以后,数组中的值呈现递减的趋势,牛牛把符合这样先增后减规律的数组定义为金字塔数组,请你帮牛牛在整个num数组中找出长度最长的金字塔数组,如果金字塔数组不存在,请输出0。
示例1
输入
4,[1,2,3,1]
输出
复制
4
示例2
输入
5,[1,5,3,3,1]
输出
3
备注:
1<=n<=1000000,且num数组中的数 0<=num[i]<=1000000。
题目大意:
要求输出金字塔数组的最大长度,所谓金字塔数组,就是先严格递增,然后再严格递减。
解题思路:
用两个数组维护,一个数组表示:左边的数比当前数小的有几个(必须是连续),另一个数组表示右边的数比当前数小的有几个(也是连续),然后去遍历0-n,如果ai 和 bi 都不为0,说明这个数是塔顶,ans=max(ai+bi+1,ans)即可维护出答案。AC代码:
class Solution {
public:
/**
*
* @param n int整型
* @param num int整型vector
* @return int整型
*/
int a[1000050],b[1000050];
int getMaxLength(int n, vector<int>& num) {
memset(a,0,sizeof a);
memset(b,0,sizeof b);
for(int i=1;i<n;i++)//维护左边比自己小的连续的有几个
if(num[i]>num[i-1])
a[i]=a[i-1]+1;
for(int i=n-2;i>=0;i--)//维护右边的比自己小的连续的有几个
if(num[i]>num[i+1])
b[i]=b[i+1]+1;
int ans=0;
for(int i=0;i<n;i++)
if(a[i]&&b[i])//判断塔顶
ans=max(ans,a[i]+b[i]+1);
return ans;
}
};