组队井字游戏

组队井字游戏

题目

Farmer John 有 26 头奶牛，恰好她们名字都以不同的字母开头，所以 Farmer John 用每头奶牛的名字的首字母来指代她——一个 A…Z 之间的字母。

这些奶牛最近沉迷于井字游戏，但是由于她们并不满足只有两头奶牛可以一起玩，她们改编了这个游戏，可以让许多奶牛可以一块儿玩！

就像常规的井字游戏一样，这个游戏是在一块 3×3 的棋盘上进行的，只是与仅用 X 和 O 不同，每个格子用一个 A…Z 之间的字母标记，表示占有这个格子的奶牛名字的首字母。

以下是一个棋盘的例子：

COW
XXO
ABC

这些奶牛会在她们迷惑于如何判断胜负之前就占满这九个格子。

显然，就像常规的井字游戏一样，如果任何一头奶牛占有了一整行、一整列，或是一整条对角线，那么这头奶牛就获胜了。

然而，由于奶牛认为多牛游戏中这并不容易出现，所以她们决定允许奶牛组成两牛一队，如果某一行、一列，或是一条对角线仅包含一队的两头奶牛的字母，并且同时包含了这两头奶牛（不仅仅是一头）的字母，那么这一队就获胜。

请帮助奶牛们判断有多少头奶牛或是两头奶牛组成的队伍可以获胜。

注意棋盘上的同一个格子可能在不同奶牛或队伍的获胜中均被用到。

输入格式

输入包含三行，每行三个 A…Z 之间的字符。

输出格式

输出包含两行。

第一行，输出能够获胜的单独的奶牛的数量。

第二行，输出能够获胜的两头奶牛组成的队伍的数量。

输入样例：

COW
XXO
ABC

输出样例：

0
2

样例解释

在这个例子中，没有单独的奶牛可以获得胜利。

但是，如果奶牛 C 和奶牛 X 组队，她们可以通过 C−X−C 对角线获胜。

同样地，如果奶牛 X 和 O 组队，她们可以通过中间一行取胜。

思路

枚举所有的行、列、对角线

然后看一下每一行、每一列、每一个对角线上到底有几个字母

如果只有1个字母，累积到第一个答案里边

如果有2个字母，累积到第二个答案里边

AC代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<set>
#include<vector>

using namespace std;

set<set<char>> ans[2];
string g[3];

void insert(vector<vector<int>> ps)
{
	set<char> S;
	for(auto& p:ps)
	{
		int x=p[0],y=p[1];
		S.insert(g[x][y]);
	}
	
	if(S.size()==1)
	{
		ans[0].insert(S);
	}
	else if(S.size()==2)
	{
		ans[1].insert(S);
	}
}

int main()
{
	for(int i=0;i<3;i++)
	{
		cin>>g[i];
	}
	
	for(int i=0;i<3;i++)
	{
		insert({{i,0},{i,1},{i,2}});
	}
	for(int i=0;i<3;i++)
	{
		insert({{0,i},{1,i},{2,i}});
	}
	
	insert({{0,0},{1,1},{2,2}});
	insert({{2,0},{1,1},{0,2}});
	
	cout<<ans[0].size()<<endl;
	cout<<ans[1].size()<<endl;
	
	return 0;
}