프로그래머스 - 혼자서 하는 틱택토

문제 설명

틱택토는 두 사람이 하는 게임으로 처음에 3x3의 빈칸으로 이루어진 게임판에 선공이 "O", 후공이 "X"를 번갈아가면서 빈칸에 표시하는 게임입니다. 가로, 세로, 대각선으로 3개가 같은 표시가 만들어지면 같은 표시를 만든 사람이 승리하고 게임이 종료되며 9칸이 모두 차서 더 이상 표시를 할 수 없는 경우에는 무승부로 게임이 종료됩니다.

할 일이 없어 한가한 머쓱이는 두 사람이 하는 게임인 틱택토를 다음과 같이 혼자서 하려고 합니다.

• 혼자서 선공과 후공을 둘 다 맡는다.
• 틱택토 게임을 시작한 후 "O"와 "X"를 혼자서 번갈아 가면서 표시를 하면서 진행한다.

틱택토는 단순한 규칙으로 게임이 금방 끝나기에 머쓱이는 한 게임이 종료되면 다시 3x3 빈칸을 그린 뒤 다시 게임을 반복했습니다. 그렇게 틱택토 수 십 판을 했더니 머쓱이는 게임 도중에 다음과 같이 규칙을 어기는 실수를 했을 수도 있습니다.

• "O"를 표시할 차례인데 "X"를 표시하거나 반대로 "X"를 표시할 차례인데 "O"를 표시한다.
• 선공이나 후공이 승리해서 게임이 종료되었음에도 그 게임을 진행한다.

게임 도중 게임판을 본 어느 순간 머쓱이는 본인이 실수를 했는지 의문이 생겼습니다. 혼자서 틱택토를 했기에 게임하는 과정을 지켜본 사람이 없어 이를 알 수는 없습니다. 그러나 게임판만 봤을 때 실제로 틱택토 규칙을 지켜서 진행했을 때 나올 수 있는 상황인지는 판단할 수 있을 것 같고 문제가 없다면 게임을 이어서 하려고 합니다.

머쓱이가 혼자서 게임을 진행하다 의문이 생긴 틱택토 게임판의 정보를 담고 있는 문자열 배열 board가 매개변수로 주어질 때, 이 게임판이 규칙을 지켜서 틱택토를 진행했을 때 나올 수 있는 게임 상황이면 1을 아니라면 0을 return 하는 solution 함수를 작성해 주세요.

---

제한사항

• board의 길이 = board[i]의 길이 = 3
  • board의 원소는 모두 "O", "X", "."으로만 이루어져 있습니다.
• board[i][j]는 i + 1행 j + 1열에 해당하는 칸의 상태를 나타냅니다.
  • "."은 빈칸을, "O"와 "X"는 해당 문자로 칸이 표시되어 있다는 의미입니다.

---

입출력 예

board	result
["O.X", ".O.", "..X"]	1
["OOO", "...", "XXX"]	0
["...", ".X.", "..."]	0
["...", "...", "..."]	1

---

내 코드

def solution(board):
    answer = 0
    cnt_O, cnt_X, win_O, win_X = 0,0,0,0
    
    for b in board:
        cnt_O += b.count('O')
        cnt_X += b.count('X')
    
    # .만 있을때
    if cnt_O + cnt_X == 0:
        return 1
    # O,X 번갈아 가는 규칙 및 게임 끝나는 룰 어겼을때 
    if cnt_O > cnt_X + 1:
        return 0
    if cnt_O < cnt_X:
        return 0
    
    # 가로,세로,대각선 같은 모양으로 3개 이루면 1씩 추가
    for i in range(3):
        if board[0][i] == board[1][i] and board[1][i] == board[2][i]:
            if board[1][i] == 'O':
                win_O += 1
            elif board[1][i] == 'X':
                win_X += 1
        if board[i][0] == board[i][1] and board[i][1] == board[i][2]:
            if board[i][1] == 'O':
                win_O += 1
            elif board[i][1] == 'X':
                win_X += 1
    if board[0][0] == board[1][1] and board[1][1] == board[2][2]:
        if board[1][1] == 'O':
            win_O += 1
        elif board[1][1] == 'X':
            win_X += 1
    if board[2][0] == board[1][1] and board[1][1] == board[0][2]:
        if board[1][1] == 'O':
            win_O += 1
        elif board[1][1] == 'X':
            win_X += 1
    
    # O와 X는 동시에 이길 수 없음
    if win_O == win_X and win_O == 0:
        answer = 1
    # O가 이겼을때: O의 개수가 X보다 많아야 한다(O가 선공)
    if win_X == 0 and win_O > 0:
        if cnt_O > cnt_X:
            answer = 1
    # X가 이겼을때 : X의 개수와 O의 개수가 같아야한다.(O부터 선공)
    if win_O == 0 and win_X > 0:
        if cnt_X == cnt_O:
            answer = 1
    
    return answer

시뮬레이션 문제로, 여러가지 case를 모두 다 따져야했다.