[백준] 2072 오목 (Python/파이썬)

<문제 링크>

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

2072번: 오목

 

<풀이 전략>

1. 오목 여부 검증 함수를 만들어서 수를 둘 때마다 검증하면 된다

2. 행, 열, 두 대각선에 대해 구현하면 된다.

3. 육목이랑 0 index 조심하기

4. bfs나 dfs 혹은 그냥 델타 탐색으로도 풀 수 있는데 생각하기 귀찮아서 빡구현으로 풀었다(코드는 전자가 더 깔끔하다)

 

<정답 코드>

import sys, os, io, atexit

input = lambda: sys.stdin.readline().rstrip('\r\n')
stdout = io.BytesIO()
sys.stdout.write = lambda s: stdout.write(s.encode("ascii"))
atexit.register(lambda: os.write(1, stdout.getvalue()))

# 221224 2072 오목

# 정답코드


# 오목 여부 확인 함수
def check_five(r, c):
    
    num = board[r][c]

    # 행 검증
    max_len = 0
    cnt = 0
    for i in range(r - 4, r + 5):
        if 0 <= i < 19:
            if board[i][c] == num:
                cnt += 1
                max_len = max(cnt, max_len)
            else:
                cnt = 0

    if max_len == 5:
        return True

    # 열 검증
    max_len = 0
    cnt = 0
    for i in range(c - 4, c + 5):
        if 0 <= i < 19:
            if board[r][i] == num:
                cnt += 1
                max_len = max(cnt, max_len)
            else:
                cnt = 0

    if max_len == 5:
        return True

    # 좌상 -> 우하 대각선
    max_len = 0
    cnt = 0
    for i in range(-4, 5):
        if 0 <= r + i < 19 and 0 <= c + i < 19:
            if board[r + i][c + i] == num:
                cnt += 1
                max_len = max(cnt, max_len)
            else:
                cnt = 0
    
    if max_len == 5:
        return True

    # 우상 -> 좌하 대각선
    max_len = 0
    cnt = 0
    for i in range(-4, 5):
        if 0 <= r + i < 19 and 0 <= c - i < 19:
            if board[r + i][c - i] == num:
                cnt += 1
                max_len = max(cnt, max_len)
            else:
                cnt = 0

    if max_len == 5:
        return True

    # 오목 아닐 시 False 반환
    return False
    


# board: 오목판
board = [[0 for _ in range(19)] for _ in range(19)]

# N: 오목 돌 둔 횟수
N = int(input())

for i in range(1, N + 1):

    # r, c: 바둑돌 위치
    r, c = map(int, input().split())
    # 0 인덱스 변환
    r -= 1
    c -= 1

    if i % 2:
        board[r][c] = 1
    else:
        board[r][c] = 2

    # 오목이 될 경우 수 출력 후 종료
    if check_five(r, c):
        print(i)
        exit()

# 승패 안 난 경우
print(-1)