[python] 백준 - 2615. 오목

🤔문제 해결

• 6목은 안된다.
• 답을 출력할 때, 왼쪽 우선, 위쪽 우선 이므로
• 바둑돌을 기준으로 우상, 우, 우하, 하 이렇게 4방향만 고려해줬다.

1. 먼저 바둑돌 하나를 선택하고
   1. 이 바둑돌이 끝에서 시작하는지 or 끝이 아니라면 내가 체크할 방향의 반대방향으로 상대 바둑돌이 없는지
      1. 육목 방지
   2. 위의 조건에 맞다면 내가 정한 방향으로 탐색하며 같은 색의 바둑돌을 계속 찾음
   3. 상대 바둑돌을 만나거나 바둑판 밖으로 나가면 내가 찾은 바둑돌의 개수를 체크
   4. 바둑돌을 4개 더 찾았으면 성공
   5. 아니면 실패

 

 

 

💻소스 코드

 import sys

input = sys.stdin.readline
LENGTH = 19

board = [list(map(int, input().split(" "))) for _ in range(19)]

# 우상, 우, 우하, 하 - 3가지 방향만 고려
dx, dy = [-1, 0, 1, 1], [1, 1, 1, 0]


def is_in(x, y):
    if 0 <= x < LENGTH and 0 <= y < LENGTH:
        return True
    else:
        return False


def is_omok(x, y, player):
    for k in range(len(dx)):
        # 반대방향에 나랑 같은 돌이 있는 지 확인: 없어야된다.
        bx = x - dx[k]
        by = y - dy[k]
        if not is_in(bx, by) or board[bx][by] != player:
            nx = x + dx[k]
            ny = y + dy[k]
            cnt = 1
            while True:
                if is_in(nx, ny) and board[nx][ny] == player:
                    cnt += 1
                else:
                    break
                nx += dx[k]
                ny += dy[k]
            if cnt == 5:
                return True
    else:
        return False


def main():
    for i in range(LENGTH):
        for j in range(LENGTH):
            if board[i][j] != 0:
                # 오목인지 점검하기
                if is_omok(i, j, board[i][j]):
                    return board[i][j], f'{i + 1} {j + 1}'
    else:
        return 0, 0


winner, pos = main()
if winner != 0:
    print(winner)
    print(pos)
else:
    print(winner)

 

📕문제 확인

출처: BACKJOON ONLINE JUDGE