4615. 재미있는 오셀로 게임

https://swexpertacademy.com/main/code/problem/problemDetail.do?contestProbId=AWQmA4uK8ygDFAXj&categoryId=AWQmA4uK8ygDFAXj&categoryType=CODE&problemTitle=%EC%98%A4%EC%85%80%EB%A1%9C&orderBy=FIRST_REG_DATETIME&selectCodeLang=ALL&select-1=&pageSize=10&pageIndex=1 

SW Expert Academy

T = int(input())
for tc in range(1, T + 1):
    N, M = map(int, input().split())
    arr = [[0] * (N + 1) for _ in range(N + 1)] # 보드 빈자리 0으로 채우기
    m = N // 2
    arr[m][m] = arr[m + 1][m + 1] = 2  # 처음 세팅
    arr[m][m + 1] = arr[m + 1][m] = 1
 
    for _ in range(M):
        si, sj, color = map(int, input().split())
        arr[si][sj] = color  # 맨처음 배치
        for di, dj in ((1, 0), (0, 1), (-1, 0), (0, -1), (1, -1), (1, 1), (-1, -1), (-1, 1)):
            rst = []
            for k in range(1, N):
                ni, nj = si + di * k, sj + dj * k
                if 1 <= ni <= N and 1 <= nj <= N:
                    if arr[ni][nj] == 0:  # 쭉 탐색하다 0이 나오면 정지
                        break
                    elif arr[ni][nj] == color:  # 본인색이 나오면 
                        while rst:  # 사이에 상대색이 있으면? 모두 본인색이 될 때까지
                            ti, tj = rst.pop()  # 그걸 빼고
                            arr[ti][tj] = color # 본인색으로 바꾸기
                        break
                    else:
                        rst.append((ni, nj)) # 상대색이 나오면 rst에 저장
                else:
                    break
    black = 0
    white = 0
 
    for lst in arr:  # 전체에서 해당 색 각각 카운트
        black += lst.count(1)
        white += lst.count(2)
    print(f'#{tc} {black} {white}')