Team Tic Tac Toe

传送门

题面描述：

给出一个3*3井字棋一样的图，图中的9个格由 A 到 Z 的字母组成。若三个相同字母连成一线则该字母个人获胜，若两种不同字母练成一线则由这两种字母组成的队伍获胜，求对于给出的图分别有多少种个体和组队的获胜方法。

题目分析：

由于只有3*3的大小，所以我是直接暴力遍历了，但是也有几个值得注意的点。

1. 每个个体最多只能获胜一次
2. 每个队伍最多也只能获胜一次
3. 每个个体可以和其他任意个体组成不同的队伍，以此达到的获胜也成立

例如下面这种情况：
A A A
A X C
A X C
答案则是：个体胜利 1（A）；组队胜利 2（AX,AC）
为了避免重复我是多开了一个 single 数组保存个体和 team 数组保存组队。
这题代码写的是真的丑…

代码：

#include<algorithm>
#include<stdio.h> 
#include<iostream>
#include<string.h> 
#include<queue>
#include<vector>
using namespace std;
int i,j,single[26],team[26][26],b[26],ans1=0,ans2=0;
char a[3][3];

int main(){
	for(i=0;i<3;i++) cin>>a[i];
	for(i=0;i<3;i++){
		b[a[i][0]-'A']++,b[a[i][1]-'A']++,b[a[i][2]-'A']++;
		for(j=0;j<26;j++){
			if(b[j]==3) single[j]=1;
			else if(b[j]==2) for(int k=0;k<26;k++) if(b[k]==1) team[j][k]=1,team[k][j]=1;
		}
		memset(b,0,sizeof(b));
	}
	for(i=0;i<3;i++){
		b[a[0][i]-'A']++,b[a[1][i]-'A']++,b[a[2][i]-'A']++;
		for(j=0;j<26;j++){
			if(b[j]==3) single[j]=1;
			else if(b[j]==2) for(int k=0;k<26;k++) if(b[k]==1) team[j][k]=1,team[k][j]=1;
		}
		memset(b,0,sizeof(b));
	}
	b[a[0][0]-'A']++,b[a[1][1]-'A']++,b[a[2][2]-'A']++;
	for(j=0;j<26;j++){
	    if(b[j]==3) single[j]=1;
	    else if(b[j]==2) for(int k=0;k<26;k++) if(b[k]==1) team[j][k]=1,team[k][j]=1;
	}
	memset(b,0,sizeof(b));
	b[a[0][2]-'A']++,b[a[1][1]-'A']++,b[a[2][0]-'A']++;
	for(j=0;j<26;j++){
	    if(b[j]==3) single[j]=1;
	    else if(b[j]==2) for(int k=0;k<26;k++) if(b[k]==1) team[j][k]=1,team[k][j]=1;
	}
	for(i=0;i<26;i++) if(single[i]==1) ans1++;
	for(i=0;i<26;i++) for(j=i;j<26;j++) if(team[i][j]==1) ans2++;
	cout<<ans1<<endl<<ans2;
}