[SWEA 코딩] 틱택톰 - PYTHON #11545

소스 코드

T = int(input())

# 출력값 저장
res_list = ['O won', 'X won', 'Draw', 'Game has not completed']

# 출력에 필요한 index 값 저장
my_items = []

for tc in range(T):

    # 주어진 input이 T-1이 전에는 5줄임
    if tc < T - 1:
        board = [list(input()) for _ in range(5)]

    else:
        board = [list(input()) for _ in range(4)]

    # '.'이 없으면 0 있으면 1
    check = 0

    # 이긴 사람이 있으면 1 없으면 0
    res = 0

    # 행,열 확인
    for q in range(4):
        temp_row = 0
        temp_col = 0
        cnt_row = 0
        cnt_col = 0
        for w in range(4):

            # 행의 값이 변하지 않았으면
            if temp_row == 0:

                # 해당 index의 값이 X 나 O이면 cnt + 1 해주고, temp에 저장
                if board[q][w] == 'X' or board[q][w] == 'O':
                    cnt_row += 1
                    temp_row = board[q][w]

                # '.'이면 check 바꾸기
                elif board[q][w] == '.':
                    check = 1

                # T이면 cnt + 1 만
                else:
                    cnt_row += 1

            # 행의 값에 X or O가 저장된 상태면
            else:

                # 현재 index의 값이 temp에 저장된 값이랑 같거나 T이면
                if (board[q][w] == temp_row) or board[q][w] == 'T':
                    cnt_row += 1

                elif board[q][w] == '.':
                    check = 1

            if temp_col == 0:
                if board[w][q] == 'X' or board[w][q] == 'O':
                    cnt_col += 1
                    temp_col = board[w][q]

                elif board[w][q] == '.':
                    check = 1

                else:
                    cnt_col += 1

            else:
                if (board[w][q] == temp_col) or board[w][q] == 'T':
                    cnt_col += 1

                elif board[w][q] == '.':
                    check = 1

        # cnt가 4인 경우, 저장된 값에 따라 my_items에 저장
        if cnt_row == 4:
            if temp_row == 'O':
                my_items.append(0)

            else:
                my_items.append(1)

            # res값 변경
            res = 1
            break

        if cnt_col == 4:
            if temp_col == 'O':
                my_items.append(0)

            else:
                my_items.append(1)

            res = 1
            break

    # 대각선 확인
    if res == 0:
        temp_cross1 = 0
        temp_cross2 = 0
        cnt_cross1 = 0
        cnt_cross2 = 0
        for qq in range(4):
            if temp_cross1 == 0:
                if board[qq][qq] == 'O' or board[qq][qq] == 'X':
                    cnt_cross1 += 1
                    temp_cross1 = board[qq][qq]

                elif board[qq][qq] == 'T':
                    cnt_cross1 += 1

            else:
                if (board[qq][qq] == temp_cross1) or board[qq][qq] == 'T':
                    cnt_cross1 += 1

            if temp_cross2 == 0:
                if board[qq][3 - qq] == 'O' or board[qq][3 - qq] == 'X':
                    cnt_cross2 += 1
                    temp_cross2 = board[qq][3 - qq]

                elif board[qq][3 - qq] == 'T':
                    cnt_cross2 += 1

            else:
                if (board[qq][3 - qq] == temp_cross2) or board[qq][3 - qq] == 'T':
                    cnt_cross2 += 1

        if cnt_cross1 == 4:
            if temp_cross1 == 'O':
                my_items.append(0)

            else:
                my_items.append(1)

            res = 1

        if cnt_cross2 == 4:
            if temp_cross2 == 'O':
                my_items.append(0)

            else:
                my_items.append(1)

            res = 1

    if res == 0:
        
        # '.'이 없었으면
        if check == 0:
            
            # Draw
            my_items.append(2)

        else:
            
            # 그 외
            my_items.append(3)

idx = 1
for r in my_items:
    print('#{} {}'.format(idx, res_list[r]))
    idx += 1

 

---

해결 방법

1. 행 4개 열 4개 대각선 2개 케이스마다 총 10개만 확인할 수 있는 코드를 짜면 된다

 

2. 최대한 연산을 줄이기 위해 res값을 저장해서 불필요한 연산을 방지함

---

느낀 점

1. 속도도 좋지만 코드를 단순하게 짤 수 있는 아이디어도 중요하다

 

ps. 개인적인 코드와 코드를 작성의 이유를 적은 것입니다.

 

오류나 적절치 않은 문법이 존재할 수 있으며, 다른 분들께는 굉장히 비효율적인 방법으로 여겨질 수 있습니다.

 

혹시 개선 사항, 오류 및 문제에 대한 수정 사항 등을 댓글로 남겨주신다면 감사한 마음으로 배우고 수정하겠습니다.