백준/ Silver 1 문제 , 백준 파이썬 1303 , 전쟁 - 전투 [BFS]

백준/ Silver 1 문제 , 백준 파이썬 1303 , 전쟁 - 전투 [BFS]

 

 

 

 

Check Point !   ( 해당사항 ✓체크 )

1. 막힘 없이 수월하게 풀린 문제인가?

2. 1시간이내로 풀렸던 문제인가?

3. 1시간 이상 or 며칠을 두고 풀어봤더니 풀린 문제인가?

4. 시간을 써도 도무지 풀 수 없는 문제인가?

5. 솔루션을 찾아봤는가?

-------------------------------------------------------------------------------------------
난이도 체감
1. 최상
2. 상
3. 중
4. 하


<이해도>
1. 완벽히 이해
2. 다소 헷갈리는 부분들이 있음
3. 이해 못함


<덧붙일 말>
자꾸 틀려서 뭔가 계속 봤는데 N이 세로가 아니고 가로였다 ㅡ,ㅡ

 

 

문제 출처

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

1303번: 전쟁 - 전투

 

 

 

 

 

정답

 

 

from collections import deque
import sys
input=sys.stdin.readline

vertical=[1,-1,0,0]
parallel=[0,0,1,-1]

M,N=map(int,input().split())
MAP=[list(input()) for _ in range(N)]

def bfs(x,y,a):
    dq=deque()
    dq.append([x,y])
    MAP[x][y]=0
    cnt=1
    while dq:
        X,Y=dq.popleft()
        for i in range(4):
            mx,my=X+vertical[i],Y+parallel[i]
            if 0<=mx<N and 0<=my<M:
                if MAP[mx][my]==a:
                    MAP[mx][my]=0
                    cnt+=1
                    dq.append([mx,my])
    return cnt**2

answer=[0,0]
for i in range(N):
    for j in range(M):
        if MAP[i][j]=='W':
            answer[0]+=bfs(i,j,'W')
        elif MAP[i][j]=='B':
            answer[1]+=bfs(i,j,'B')

print(*answer)