WUST OJ 1540 小明爱数组

1540: 小明爱数组

• Description

小明最近喜欢上了研究数组，他想得到一个先递增后递减的数组序列，并且递增的序列的元素个数与递减的序列的元素个数相等，而且递增的序列和递减的序列都是等差数列。给出一个数组序列，问是否是小明想要的。

• Input

多组测试，第一行是一个N(N是奇数，1<=N<=99),第二行是N个整数(int范围以内)，代表数组序列。

• Output

如果是小明想要的，输出666，否则输出TNT。

• Sample Input

5
1 3 5 4 3
7
1 2 3 4 5 6 7

• Sample Output

666
TNT

分析
这题需要判断递增的元素和递减的元素个数是否相等，所以我们直接找到中间一项，分别对前半部分和后半部分进行遍历，判断是否符合规定即可。但这个题有个大坑，我踩了半天才跳出来，N等于1的时候是符合要求的，所以直接输出666

#include<iostream>
using namespace std;
int judge(int *a,int n)
{
    int i,d1,d2;
    d1=a[1]-a[0];   //递增数列的公差d1
    d2=a[n/2]-a[(n+1)/2];    //递减数列的公差d2
    if(d1<=0||d2<=0)
        return 0;
    for(i=1;i<=n/2;i++)
        if(a[i]-a[i-1]!=d1)
        return 0;
    for(i=n/2;i<n-1;i++)
        if(a[i]-a[i+1]!=d2)
        return 0;
    return 1;
}
int main()
{
    int n,a[100],i;
    while(cin>>n)
    {
        if(n==1)
            {
                cin>>a[0];
                cout<<"666"<<'\n';
                continue;
            }
        int flag=1;
       for(i=0;i<n;i++)
            cin>>a[i];
        if(judge(a,n))
            cout<<"666"<<endl;
        else
            cout<<"TNT"<<'\n';
    }
    return 0;
}