문제
미키의 뒷마당에는 특정 수의 양이 있다. 그가 푹 잠든 사이에 배고픈 늑대는 마당에 들어와 양을 공격했다.
마당은 행과 열로 이루어진 직사각형 모양이다. 글자 ‘.’ (점)은 빈 필드를 의미하며, 글자 ‘#’는 울타리를, ‘o’는 양, ‘v’는 늑대를 의미한다.
한 칸에서 수평, 수직만으로 이동하며 울타리를 지나지 않고 다른 칸으로 이동할 수 있다면, 두 칸은 같은 영역 안에 속해 있다고 한다. 마당에서 “탈출”할 수 있는 칸은 어떤 영역에도 속하지 않는다고 간주한다.
다행히 우리의 양은 늑대에게 싸움을 걸 수 있고 영역 안의 양의 수가 늑대의 수보다 많다면 이기고, 늑대를 우리에서 쫓아낸다. 그렇지 않다면 늑대가 그 지역 안의 모든 양을 먹는다.
맨 처음 모든 양과 늑대는 마당 안 영역에 존재한다.
아침이 도달했을 때 살아남은 양과 늑대의 수를 출력하는 프로그램을 작성하라.
입력
첫 줄에는 두 정수 R과 C가 주어지며(3 ≤ R, C ≤ 250), 각 수는 마당의 행과 열의 수를 의미한다.
다음 R개의 줄은 C개의 글자를 가진다. 이들은 마당의 구조(울타리, 양, 늑대의 위치)를 의미한다.
출력
하나의 줄에 아침까지 살아있는 양과 늑대의 수를 의미하는 두 정수를 출력한다.
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
import sys
sys.setrecursionlimit(100000)
R,C = map(int,input().split())
yard = [list(input()) for _ in range(R)]
dx = [-1,1,0,0]
dy = [0,0,-1,1]
total_v = 0
total_o = 0
o = 0
v = 0
def dfs(x,y):
global v,o
# 개수 세기
for i in range(4):
nx = x + dx[i]
ny = y + dy[i]
if -1<nx<R and -1<ny<C and yard[nx][ny]!='#':
if yard[nx][ny] == 'v':
v+=1
elif yard[nx][ny] == 'o':
o+=1
yard[nx][ny]='#' #방문처리
dfs(nx,ny)
return
for x in range(R):
for y in range(C):
if yard[x][y] != '#':
if yard[x][y] == 'v':
v+=1
elif yard[x][y] == 'o':
o+=1
yard[x][y] = '#'
dfs(x,y)
if o>v:
total_o+=o
else:
total_v+=v
o = 0
v = 0
print(total_o,total_v)
풀이방식
방문하는 기준은 # 샵이 아닐 경우이다.
방문하면 해당 위치를 #으로 만든다 (visited를 만들 필요 없음)
해당 위치의 값이 v인 경우 v의 개수를 세고, o인 경우 o의 개수를 센다.
결국 dfs가 return될 때마다 해당 영역의 v 수와 o 수를 알 수 있고, 개수를 비교해서 o가 v보다 더 많을 때에만 전체 양 수에 o를 더해준다. 반대로 그 외 모든 경우에는 전체 늑대 수에 v를 더해준다.