Problem: [Usaco2018 Open]Team Tic Tac Toe

Problem: [Usaco2018 Open]Team Tic Tac Toe

Time Limit: 1 Sec Memory Limit: 128 MB

Description

Farmer John有26头奶牛，恰好她们名字都以不同的字母开头，所以Farmer John用每头奶牛的名字的首字母来指代她——一个A…Z之间的字母。这些奶牛最近沉迷于井字游戏，但是由于她们并不满足只有两头奶牛可以一起玩，她们改编了这个游戏，可以让许多奶牛可以一块儿玩！就像常规的井字游戏一样，这个游戏是在一块3×3的棋盘上进行的，只是与仅用X和O不同，每个格子用一个A…Z之间的字母标记，表示占有这个格子的奶牛名字的首字母。以下是一个棋盘的例子：
COW
XXO
ABC
这些奶牛会在她们迷惑于如何判断胜负之前就占满这九个格子。显然，就像常规的井字游戏一样，如果任何一头奶牛占有了一整行、一整列，或是一整条对角线，那么这头奶牛就获胜了。然而，由于奶牛认为多牛游戏中这并不容易出现，所以她们决定允许奶牛组成两牛一队，如果某一行、一列，或是一条对角线仅包含一队的两头奶牛的字母，并且同时包含了这两头奶牛（不仅仅是一头）的字母，那么这一队就获胜。
请帮助奶牛们判断有多少头奶牛或是两头奶牛组成的队伍可以获胜。
注意棋盘上的同一个格子可能在不同奶牛或队伍的获胜中均被用到。

Input

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

Output

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

Sample Input

COW
XXO
ABC

Sample Output

0
2
HINT
在这个例子中，没有单独的奶牛可以获得胜利。
但是，如果奶牛C和奶牛X组队，她们可以通过C-X-C对角线获胜。
同样地，如果奶牛X和O组队，她们可以通过中间一行取胜
代码如下

#include <iostream>
using namespace std;
 
char B[3][3];
 
// Does 1 cow win?
int cow_wins(char ch) {
    // Check diagonals
    if (B[0][0] == ch && B[1][1] == ch && B[2][2] == ch) return 1;
    if (B[0][2] == ch && B[1][1] == ch && B[2][0] == ch) return 1;
 
    // Check rows and columns
    for (int i=0; i<3; i++) {
        if (B[0][i] == ch && B[1][i] == ch && B[2][i] == ch) return 1;
        if (B[i][0] == ch && B[i][1] == ch && B[i][2] == ch) return 1;
    }
 
    return 0;
}
 
// Test if a team wins based on 3 characters in a row, column, or diagonal
bool check3(char ch1, char ch2, char a, char b, char c) {
    // All 3 characters have to be either ch1 or ch2
    if (a != ch1 && a != ch2) return false;
    if (b != ch1 && b != ch2) return false;
    if (c != ch1 && c != ch2) return false;
 
    // ch1 and ch2 have to appear at least once each
    if (a != ch1 && b != ch1 && c != ch1) return false;
    if (a != ch2 && b != ch2 && c != ch2) return false;
 
    return true;
}
 
// Does a team win?
int team_wins(char ch1, char ch2) {
    // Check diagonals
    if (check3(ch1, ch2, B[0][0], B[1][1], B[2][2])) return 1;
    if (check3(ch1, ch2, B[0][2], B[1][1], B[2][0])) return 1;
 
    // Check rows and columns
    for (int i=0; i<3; i++) {
        if (check3(ch1, ch2, B[0][i], B[1][i], B[2][i])) return 1;
        if (check3(ch1, ch2, B[i][0], B[i][1], B[i][2])) return 1;
    }
 
    return 0;
}
 
int main(void) {
    for (int i=0; i<3; i++)
        for (int j=0; j<3; j++)
            cin >> B[i][j];
 
    int answer1 = 0, answer2 = 0;
    for (char ch = 'A'; ch <= 'Z'; ch++)
        answer1 += cow_wins(ch);
    for (char ch1 = 'A'; ch1 <= 'Z'; ch1++)
        for (char ch2 = ch1+1; ch2 <= 'Z'; ch2++)
            answer2 += team_wins(ch1, ch2);
 
    cout << answer1 << "
" << answer2 << "
";
    return 0;
}