티스토리 뷰

코딩/파이썬 백준

백준 1303 전쟁-전투 (파이썬)

박상어 2022. 2. 3. 17:14
728x90

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

 

1303번: 전쟁 - 전투

첫째 줄에는 전쟁터의 가로 크기 N, 세로 크기 M(1 ≤ N, M ≤ 100)이 주어진다. 그 다음 두 번째 줄에서 M+1번째 줄에는 각각 (X, Y)에 있는 병사들의 옷색이 띄어쓰기 없이 주어진다. 모든 자리에는

www.acmicpc.net

 

 

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
from collections import deque
 
 
def bfs(x, y, color):
    cnt = 1
    dq = deque()
    dq.append([x, y])
    visit[x][y] = True
    while dq:
        x, y = dq.popleft()
        for dx, dy in direction:
            nx, ny = x + dx, y + dy
            if 0 <= nx < M and 0 <= ny < N and war_map[nx][ny] == color and not visit[nx][ny]:
                dq.append([nx, ny])
                visit[nx][ny] = True
                cnt += 1
    return cnt * cnt
 
 
N, M = map(int, input().split())
direction = [(0, 1), (1, 0), (0, -1), (-1, 0)]
visit = [[False for _ in range(N)] for _ in range(M)]
war_map = []
W, B = 0, 0
for i in range(M):
    war_map.append(list(input().rstrip()))
for i in range(M):
    for j in range(N):
        if war_map[i][j] == 'W' and not visit[i][j]:
            W += bfs(i, j, war_map[i][j])
        elif war_map[i][j] == 'B' and not visit[i][j]:
            B += bfs(i, j, war_map[i][j])
print(W, B)
 cs

 

'코딩 > 파이썬 백준' 카테고리의 다른 글

백준 1926 그림 (파이썬)  (0) 2022.02.04
백준 2992 크면서 작은 수 (파이썬)  (0) 2022.02.04
백준 9935 문자열 폭발 (파이썬)  (0) 2022.02.03
백준 5585 거스름돈 (파이썬)  (0) 2022.02.03
백준 1743 음식물 피하기 (파이썬)  (0) 2022.02.02
댓글