刷题记录:牛客NC200519Tic-Tac-Toe

传送门:牛客

小 A 和小 B 在玩井字棋，他们不懂规则，所以可能出现一个人连下多手等情况。
他们希望结束游戏，请你来判定输赢。
判断小A/小B 赢的依据是存在任一行/列/对角线上的棋子都是 小A/小B 下的。
对于每组数据输出一行一个字符串，若小 A 赢，输出 Yes ，小 B 赢输出 No，两人都满足获胜条件输出 invalid ，前
面三种情况都不符合则输出 draw 。
输入:
2
AAA
AAA
...
...
BBB
AAA
输出:
Yes
invalid

主要思路:

从题面上看是一道井字形的题目,如果是求必胜策略之类的话,这道题应该也不能算上很简单,但是现在只是让我们判断已经摆好的局势,那么这便成了一道简单的模拟题,暴力照样可做(比如判断对角线的部分我就直接暴力判断了,我思考的速度跟不上我打字的速度(bushi
小技巧:使用函数可以将A和B的判断整合成一块,避免重复操作,注意还有draw的情况(当时我忘了挂了一个点好像)

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <vector>
#include <map>
#include <set>
#include <queue>
#include <string.h>
#include <stack>
using namespace std;
typedef long long ll;
#define inf 0x3f3f3f3f
#define root 1,n,1
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
inline ll read() {
	ll x=0,w=1;char ch=getchar();
	for(;ch>'9'||ch<'0';ch=getchar()) if(ch=='-') w=-1;
	for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
	return x*w;
}
#define maxn 1000000
char mp[100][100];
int dfs(string a) {
	for(int i=1;i<=3;i++) {
		int sum=0;int sum2=0;
		for(int j=1;j<=3;j++) {
			if(mp[i][j]==a[0]) sum++;
			if(mp[j][i]==a[0]) sum2++;
		}
		if(sum==3||sum2==3) return true;
	}
	if(mp[1][1]==a[0]&&mp[2][2]==a[0]&&mp[3][3]==a[0]) return true;
	if(mp[1][3]==a[0]&&mp[2][2]==a[0]&&mp[3][1]==a[0]) return true;
	return false;
}
int main() {
	int T;T=read();
	while(T--) {
		memset(mp,'\0',sizeof(mp));
		for(int i=1;i<=3;i++) {
			for(int j=1;j<=3;j++) {
				cin>>mp[i][j];
			}
		}
		bool flag1=dfs("A"),flag2=dfs("B");
		if(flag1==true) {
			if(flag2==true) printf("invalid\n");
			else printf("Yes\n");
		}
		else {
			if(flag2==true) printf("No\n");
			else printf("draw\n");
		}
	}
	return 0;
}