재미있는 오셀로 게임 Python (SW Expert Academy)

출처 : https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWQmA4uK8ygDFAXj

SW Expert Academy

1. 문제 설명

오셀로라는 게임은 흑돌과 백돌을 가진 사람이 번갈아가며 보드에 돌을 놓아서 최종적으로 보드에 자신의 돌이 많은 사람이 이기는 게임이다.

보드는 4x4, 6x6, 8x8(가로, 세로 길이) 크기를 사용한다. 6x6 보드에서 게임을 할 때, 처음에 플레이어는 다음과 같이 돌을 놓고 시작한다(B : 흑돌, W : 백돌).

4x4, 8x8 보드에서도 동일하게 정가운데에 아래와 같이 배치하고 시작한다.

그리고 흑, 백이 번갈아가며 돌을 놓는다.

처음엔 흑부터 시작하는데 이 때 흑이 돌을 놓을 수 있는 곳은 다음과 같이 4군데이다.

플레이어는 빈공간에 돌을 놓을 수 있다.

단, 자신이 놓을 돌과 자신의 돌 사이에 상대편의 돌이 있을 경우에만 그 곳에 돌을 놓을 수 있고, 그 때의 상대편의 돌은 자신의 돌로 만들 수 있다.

(여기에서 "사이"란 가로/세로/대각선을 의미한다.)

(2, 3) 위치에 흑돌을 놓은 후의 보드는 다음과 같다.

이런 식으로 번갈아가며 흑, 백 플레이어가 돌을 놓는다.

만약 돌을 놓을 곳이 없다면 상대편 플레이어가 다시 돌을 놓는다.

보드에 빈 곳이 없거나 양 플레이어 모두 돌을 놓을 곳이 없으면 게임이 끝나고 그 때 보드에 있는 돌의 개수가 많은 플레이어가 승리하게 된다.


 [입력]

첫 번째 줄에 테스트 케이스의 수 T가 주어진다.

각 테스트 케이스의 첫 번째 줄에는 보드의 한 변의 길이 N과 플레이어가 돌을 놓는 횟수 M이 주어진다. N은 4, 6, 8 중 하나이다.

그 다음 M줄에는 돌을 놓을 위치와 돌의 색이 주어진다.

돌의 색이 1이면 흑돌, 2이면 백돌이다.

만약 3 2 1이 입력된다면 (3, 2) 위치에 흑돌을 놓는 것을 의미한다.

돌을 놓을 수 없는 곳은 입력으로 주어지지 않는다.

 [출력]

각 테스트 케이스마다 게임이 끝난 후 보드 위의 흑돌, 백돌의 개수를 출력한다.

흑돌이 30개, 백돌이 34인 경우 30 34를 출력한다.

 

2. 코드

8방향으로 진행해보며 조건에 맞는 경우 돌의 색을 바꾸도록 코드를 작성하였습니다. 8개의 방향을 하나의 반복문으로 작성하려고 하다보니 수학적?인 계산이 조금 필요해서 힘들었네요... 코드자체에 큰 어려운 점은 없는 것 같습니다.

 

아래가 작성한 코드입니다.

T = int(input())
# 상, 하, 좌, 우 + 대각선 4개
dx = [0, 0, -1, 1, -1, -1, 1, 1]
dy = [-1, 1, 0, 0, -1, 1, -1, 1]
 
for t in range(1, T+1):
    N, M = map(int, input().split())
    arr = [[0 for _ in range(N)] for _ in range(N)]
    # 정가운데 지점(초기 상태 생성)
    t_N = int(N/2)
    arr[t_N-1][t_N] ,arr[t_N][t_N-1], arr[t_N-1][t_N-1] ,arr[t_N][t_N] = 1, 1, 2, 2
    for n in range(M):
        x, y, c = map(int, input().split())
        arr[y-1][x-1] = c
        # 8방향 탐색
        for p in range(8):
            cnt = 0
            cx = x-1
            cy = y-1
            # 숫자가 있는 경우 진행
            while cx+dx[p] >= 0 and cx+dx[p] < N and cy+dy[p] >= 0 and cy+dy[p] < N and arr[cy+dy[p]][cx+dx[p]]:
                # 진행하면서 같은 숫자를 만난 경우 중간에 다른 색의 돌을 바꾸고 종료
                if abs(arr[cy+dy[p]][cx+ dx[p]] - arr[y-1][x-1]) == 0:
                    for k in range(1, cnt+1):
                        cx = x-1
                        cy = y-1
                        arr[cy+dy[p]*k][cx+dx[p]*k] = c
                    break
                # 다른색의 돌을 만날 경우 cnt를 늘리며 진행
                elif arr[cy+dy[p]][cx+dx[p]] != 0 and abs(arr[cy+dy[p]][cx+dx[p]] - arr[y-1][x-1]) == 1:
                    cnt += 1
                    cy += dy[p]
                    cx += dx[p]
    sum_w = 0
    sum_b = 0
    # 검은색 돌과 흰색 돌의 개수를 체크
    for j in range(len(arr)):
        for i in range(len(arr[j])):
            if arr[j][i] == 1:
                sum_b += 1
            elif arr[j][i] == 2:
                sum_w += 1
    print('#{} {} {}'.format(t, sum_b, sum_w))