csdn高校编程挑战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>
#include<string>
using namespace std;

int main()
{
	int n;
	string input[300];
	while(cin >> n)
	{
		for(int i = 0;i < n; i++)
			cin >> input[i];
		if(n == 1)
			cout << "YES\n";
		else
		{
			bool failed = false;
			int on_diagonal = input[0][0];
			int noton_diagonal = input[0][1];
			if(on_diagonal == noton_diagonal)
				cout << "NO\n";
			else
			{
				for(int i = 0; i < n; i++)
				{
					for(int j = 0; j < n; j++)
					{
						if(i == j || i + j == n - 1)
						{
							if(input[i][j] != on_diagonal)
							{
								failed = true;
								break;
							}
						}
						else if(input[i][j] != noton_diagonal)
						{
							failed = true;
							break;
						}
					}
					if(failed)
						break;
				}
				if(failed)
					cout << "NO\n";
				else cout << "YES\n";
			}
		}
	}
}