J. Team Tic Tac Toe

- iknoom1107

문제 설명

26마리(A-Z)의 소가 틱택토를 하기 위해서 다음과 같이 룰을 변경합니다.

1. 각 칸에 O, X 대신의 각자 이름의 첫글자(A-Z)를 적습니다.

2. 개인이 아니고 팀으로도 한 줄을 완성하면 승리합니다.

변경된 룰을 적용해서 게임판이 입력으로 주어졌을 때 첫번째 줄에 1마리의 소가 승리하는 경우의 수,

두번째 줄에 2마리의 소가 승리하는 경우의 수를 출력합니다.

설명에 [1마리의 소가 승리하는 경우]가 [2마리의 소가 승리하는 경우]에 포함되는지 모호합니다.

예를 들어 XXX인 경우 A와 X가 팀이면 승리하는데 [2마리의 소가 승리하는 경우]로 세어야하는가?인데

제출해서 확인해보니 세는게 아니더군요. 따로 세어야 합니다.

문제 풀이

체크해야될 줄은 가로 3줄 / 세로 3줄 / 대각선 2줄입니다.

일단 1줄을 체크할 때마다 알파벳의 수가 1개인지 2개인지만 세면 안됩니다.

OOO

OXO

XOO

만약 예제 입력이 위와 같을 때 알파벳 개수만 세면 답은 2 6 이겠지만 답은 1 1 입니다.

그래서 중복인지 체크해야하는데 저는 set 자료구조를 썼습니다.

set 자료구조에는 중복이 허용되지 않으니 모든 경우 일단 다 집어넣고

마지막에 원소의 수만 확인하면 되니 편하죠.

소스코드

<Python>

board = [input() for _ in range(3)]
one_cow = set()
two_cow = set()

def sol(t):
    if len(t) == 1: one_cow.add(tuple(t))
    elif len(t) == 2: two_cow.add(tuple(sorted(t)))

tmp = set(board[i][i] for i in range(3))
sol(tmp)

tmp = set(board[2 - i][i] for i in range(3))
sol(tmp)

for x in board:
    tmp = set(x)
    sol(tmp)

board = [*zip(*board)] # 90도 회전

for y in board:
    tmp = set(y)
    sol(tmp)

print(len(one_cow))
print(len(two_cow))

<C++>

#include <iostream>
#include <string>
#include <set>
using namespace std;

string board[3];
set<set<char>> one_cow, two_cow;

void count(set<char> L) {
	if (L.size() == 1) one_cow.insert(L);
	if (L.size() == 2) two_cow.insert(L);
}

int main()
{
	for (int i = 0; i < 3; i++)
		cin >> board[i];
	for (int i = 0; i < 3; i++) {
		set<char> R; set<char> C;
		for (int j = 0; j < 3; j++) {
			R.insert(board[i][j]);
			C.insert(board[j][i]);
		}
		count(R); count(C);
	}

	set<char> D1; set<char> D2;
	for (int i = 0; i < 3; i++) {
		D1.insert(board[i][i]);
		D2.insert(board[i][2-i]);
	}
	count(D1); count(D2);

	cout << one_cow.size() << endl;
	cout << two_cow.size() << endl;
}