hdu 5532

input

1<=T<=2000

2<=n<=10e5

a1 a2 ... an 1<=ai<=10e5

最多二十个n>1000

output

能否从数组中移除一个数使数组变为升序或降序，能输出YES，不能输出NO

做法，用后一个数减前一个数，统计正数的个数和负数的个数，如果全部为正或全部为负肯定可以，一个正或者一个负可能可以，如果这个和前一个或者后一个相加能变成全部正或全部负便可以，因为x1=a[i]-a[i-1],x2=a[i+1]-a[i],则a[i+1]-a[i-1]=x1+x2

#include <cstdio>
#include <queue>
#include <cstring>
#include <iostream>
#include <cstdlib>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <ctime>

using namespace std;

int T,n,a,d[100010];
int main()
{
//    freopen("/home/user/桌面/in","r",stdin);
scanf("%d",&T);
while(T--)
{
int les=0,gre=0,i,t,fa=-1,fb=-1;
scanf("%d%d",&n,&a);
for(i=0;i<n-1;i++)
{
scanf("%d",&t);
d[i]=t-a;
if(fa==-1&&d[i]<0) fa=i;
if(fb==-1&&d[i]>0) fb=i;
if(d[i]==0)
{
les++;
gre++;
}
else d[i]>0?gre++:les++;
a=t;
}
//for(int i=0;i<n-1;i++) printf("%d ",d[i]);printf("\n");
if(gre==n-1||les==n-1)
{
puts("YES");
continue;
}
if(gre==n-2)
{
if(fa==0||fa==n-2||d[fa]+d[fa-1]>=0)
{
puts("YES");
continue;
}
else if(fa<n-2&&d[fa]+d[fa+1]>=0)
{
puts("YES");
continue;
}
}
if(les==n-2)
{
if(fb==0||fb==n-2||d[fb]+d[fb-1]<=0)
{
puts("YES");
continue;
}
else if(fb<n-2&&d[fb]+d[fb+1]<=0)
{
puts("YES");
continue;
}
}
puts("NO");
}
//printf("time=%.3lf",(double)clock()/CLOCKS_PER_SEC);
return 0;
}

View Code