[백준2615번] 오목 - S1/Python

https://www.acmicpc.net/problem/2615

문제

오목은 바둑판에 검은 바둑알과 흰 바둑알을 교대로 놓아서 겨루는 게임이다. 바둑판에는 19개의 가로줄과 19개의 세로줄이 그려져 있는데 가로줄은 위에서부터 아래로 1번, 2번, ... ,19번의 번호가 붙고 세로줄은 왼쪽에서부터 오른쪽으로 1번, 2번, ... 19번의 번호가 붙는다.

위의 그림에서와 같이 같은 색의 바둑알이 연속적으로 다섯 알을 놓이면 그 색이 이기게 된다. 여기서 연속적이란 가로, 세로 또는 대각선 방향 모두를 뜻한다. 즉, 위의 그림은 검은색이 이긴 경우이다. 하지만 여섯 알 이상이 연속적으로 놓인 경우에는 이긴 것이 아니다.

입력으로 바둑판의 어떤 상태가 주어졌을 때, 검은색이 이겼는지, 흰색이 이겼는지 또는 아직 승부가 결정되지 않았는지를 판단하는 프로그램을 작성하시오. 단, 검은색과 흰색이 동시에 이기거나 검은색 또는 흰색이 두 군데 이상에서 동시에 이기는 경우는 입력으로 들어오지 않는다.

입력

19줄에 각 줄마다 19개의 숫자로 표현되는데, 검은 바둑알은 1, 흰 바둑알은 2, 알이 놓이지 않는 자리는 0으로 표시되며, 숫자는 한 칸씩 띄어서 표시된다.

풀이

#팔각이동:
directions = [(1, 0), (0, 1), (1, 1), (1, -1)]
field = [list(map(int, input().split())) for _ in range(19)]
visited = [[0]*19 for _ in range(19)]
for i in range(19):
    for j in range(19):
        # 0 이 아닌 경우
        if field[i][j] != 0 and visited[i][j] == 0:
            visited[i][j] = 1
            for k in range(4):
                combo, total, l = [(i+1,j+1)], 1, 1
                # print(f'Start combo: {i,j}')
                while True:
                    next_x, next_y = i+l*directions[k][0], j + l*directions[k][1]
                    if (0 <= next_x < 19 and 0 <= next_y < 19) and field[next_x][next_y] == field[i][j]:

                        combo.append((next_x+1, next_y+1))
                        total += 1
                    else:
                        break
                    l += 1
                # print(total)
                if total == 5:
                    combo.sort(key=lambda x: (x[1], x[0]))
                    print(field[i][j])
                    print(*combo[0])
                    exit()
                elif total > 5:

                    for tx, ty in combo:
                        visited[tx-1][ty-1] = 1
print(0)

해결방법

여섯개이상 일렬로 놓인 경우는 승리할 수 없기 때문에 해당 조건 예외를 구현하는 것이 중요한 문제.
또한, 승리한 경우 오목이 두어진 위치에서 가장 왼쪽에 놓긴 것을 출력해야 하는 조건도 유의할 것.
입력 크기도 작아서 시간 초과가 발생할 수 없는 문제이지만 문제에서 제시한 조건 때문에 난이도가 높게 책정된 문제라고 볼 수 있다.