🟩 [백준] [Python] [Silver1] 1303번_전쟁-전투

문제

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

1303번: 전쟁 - 전투

풀이

# 백준 1303번 문제 - 전쟁-전투
from collections import deque
dx, dy = [0,0,-1,1], [1,-1,0,0]
n, m = map(int, input().split())
graph = [list(input()) for i in range(m)]
visited = [[0]*n for j in range(m)]

def bfs(x, y, color):
    q = deque()
    q.append((x, y))
    visited[x][y] = 1
    cnt = 1
    while q:
        x, y = q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0 <= nx < m and 0 <= ny < n and not visited[nx][ny] and graph[nx][ny] == color:
                cnt += 1
                visited[nx][ny] = 1
                q.append((nx, ny))
    return cnt**2
white_cnt = 0
blue_cnt = 0
for i in range(m):
    for j in range(n):
        if not visited[i][j]:
            if graph[i][j] == 'W':
                white_cnt += bfs(i, j, 'W')
            else:
                blue_cnt += bfs(i, j, 'B')
print(white_cnt, blue_cnt)

이번 문제는 BFS로 쉽게 해결이 가능한 문제였다.

처음의 오류는 그래프를 탐색하는 순간 cnt를 1로 두어야 하는데 0으로 두어서 값이 다르게 나왔다.

이 오류를 해결 후에도 계속 런타임 에러가 났다.

가로, 세로의 인덱스 설정을 잘못해서 오류가 났던 것이다. 문제를 잘 읽고 인덱스를 잘 확인하도록 해야겠다.