四川大学线下编程比赛第二题：Peter的X

Peter是个小男孩，他总是背不清26个英文字母。于是，刁钻的英语老师给他布置了一个非常奇怪的作业，老师给了他一个由26个英文字母构成的N*N 的矩阵（我们保证N一定是一个奇数），问他这个矩阵构成的图案是否是一个标准的“X”。 一个标准的X的定义：1、对角线上所有元素都是同一个字母。2、所有非对角线上的元素也都是同一个字母，且字母与对角线上的字母不同。 如果是则输出“YES”，否则输出“NO”。 输入描述： 多组测试数据（数据量在100组以内）。每组测试数据第一行为一个整数N（1<=N<=300），表示表格的大小。 下面是N行是一个由字母构成的N*N的矩阵。 输出描述： 对于每组输入数据，输出一行“YES”或“NO”。 

输入样例：

5

xooox

oxoxo

soxoo

oxoxo

xooox

3

wsw

sws

wsw

3

xpx

pxp

xpe

输出样例：

NO

YES

NO

#include <iostream>
#define N 301

char a[N][N];
using namespace std;
bool isX(int n);
int main()
{
    int n;
    while( cin >> n )
    {
        for( int j = 0; j < n; j ++ )
            for( int i = 0; i < n; i ++ )
                cin >> a[i][j];
        if( isX( n ) )
            cout << "YES" << endl;
        else
            cout << "NO" << endl;


    }
    return 0;
}


bool isX(int n)
{
    char t = a[0][0], s = a[0][1];
    if( ( t == s ) || ( n % 2 == 0 ) )  return false;
    for( int j = 0; j < n; j ++ )
    {
        for( int i = 0; i < n; i ++ )
        {
            if( ( i == j ) || ( i == (n - j -1 ) ) )
            {
                if( t != a[i][j] )
                    return false;
            }
            else
            {
                if( s != a[i][j] )
                    return false;
            }
        }
    }
    return true;
}