C++11(clang++ 3.9) 解法, 执行用时: 5ms, 内存消耗: 492K, 提交时间: 2020-02-16 14:32:35

#include<stdio.h>
char mat[3][4];
int win(char c)
{
	int i;
	for(i=0;i<3;i++)
	{
		if(mat[i][0]==c&&mat[i][1]==c&&mat[i][2]==c||mat[0][i]==c&&mat[1][i]==c&&mat[2][i]==c)
		return 1;
	}
	if(mat[0][0]==c&&mat[1][1]==c&&mat[2][2]==c||mat[0][2]==c&&mat[1][1]==c&&mat[2][0]==c)
	return 1;
	return 0;
}
int main()
{
	int T,i,a,b;
	scanf("%d",&T);
	while(T--)
	{
		for(i=0;i<3;i++)
		scanf("%s",mat[i]);
		a=win('A');
		b=win('B');
		if(a&&!b)
		puts("Yes");
		else if(!a&&b)
		puts("No");
		else if(a&&b)
		puts("invalid");
		else
		puts("draw");
	}
	return 0;
}

Python3(3.5.2) 解法, 执行用时: 24ms, 内存消耗: 3436K, 提交时间: 2020-02-16 12:52:09

def List(X):
    return X + [''.join([X[0][i],X[1][i],X[2][i]]) for i in range(3)] + [''.join([X[i][i]for i in range(3)]),''.join([X[i][2-i]for i in range(3)])]

T = int(input())
for t in range(T):
    X = []
    for i in range(3):
        X.append(input())
    LX = List(X)
    if 'AAA' in LX:
        if 'BBB' in LX:
            print('invalid')
        else:
            print('Yes')
    else:
        if 'BBB' in LX:
            print('No')
        else:
            print('draw')