Description
有N 个军训新学员开始了他们的军训生活,第一天教官教他们站队,根据他们的身高,站成“凸”字形或者“凹”字形,也就是从左到右身高依次递增再递减或者依次递减再递增,如果是“凸”字形队列,身高最高的人可以有一个或者多个,其他人必须严格递增或或者递减,同样地,如果是“凹”字形队列,身高最矮的人可以有一个或者多个,其他人必须严格递增或者递减,现在这N 个同学已经站好了队,请你来判断一下现在这个队形是否是“凸”的或者“凹”的,如果是即输出YES,否则输出NO。
Input
一个整数T(T<=50)表示数据组数,接下来每组数据包含两部分,首先是一个整数N(N<=100000),表示人数,然后下一行N 个整数,表示每个人的身高h(h<=100000)。
Output
对于每组数据,如果队列是“凸”的或者“凹”的,则输出YES,否则输出NO。
Sample Input
5
5
1 2 3 2 1
5
3 2 1 2 3
5
1 1 2 1 1
5
1 2 3 4 5
5
5 4 3 2 1
Sample Output
YES
YES
NO
NO
NO
HINT
凸中最高和凹中最矮可以有多个,其他的必须是严格递增递减,队形没必要是两端对称,比如1 2 3 3 2也是凸。
这个题是有bug的,我们只找了凸形中最高的起始与终止点,凹形中最低的起始点与终止点,接着便从两个点向两边扩展,但是若1 2 3 4 3 4 3 2 1是不符合条件的,若我们遍历查找起始点与终止点中间的数据便会超时
#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
int main()
{
int T;
cin>>T;
while(T--)
{
int n,maxs=-1,mins=100001;
int en1=1,en2=1,st1,st2;
cin>>n;
int A[100001];
for(int i=1; i<=n; i++)
{
scanf("%d",&A[i]);
if(A[i]>=maxs)
{
maxs=A[i]; //记录最大值的末尾
en1=i;
}
if(A[i]<=mins)
{
mins=A[i]; //记录最小值的末尾
en2=i;
}
}
for(int i=1; i<=n; i++)
if(A[i]==maxs)
{
st1=i; //找最大的起始点
break;
}
for(int i=1; i<=n; i++)
if(A[i]==mins)
{
st2=i; //找最小的起始点
break;
}
if(A[1]<A[2]&&A[n]<A[n-1])//凸
{
int i,j;
int flag1=1,flag2=1;
for(i=st1-1; i>=1; i--)
if(A[i]>=A[i+1])
{
flag1=0;
break;
}
for(j=n; j>en1; j--)
if(A[j]>=A[j-1])
{
flag2=0;
break;
}
if(flag1==1&&flag2==1)
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
else if(A[1]>A[2]&&A[n]>A[n-1])//凹
{
int i,j;
int flag1=1,flag2=1;
for(i=st2-1; i>=1; i--)
if(A[i]<=A[i+1])
{
flag1=0;
break;
}
for(j=n; j>en2; j--)
if(A[j]<=A[j-1])
{
flag2=0;
break;
}
if(flag1==1&&flag2==1)
cout<<"YES"<<endl;
else
cout<<"NO"<<endl;
}
else
{
cout<<"NO"<<endl;
}
}
return 0;
}