(Python) - BOJ(3187번) : 양치기 꿍

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

3187번: 양치기 꿍

---

• 해설 : 

양치기 꿍은 맨날 늑대가 나타났다고 마을 사람들을 속였지만 이젠 더이상 마을 사람들이 속지 않는다. 화가 난 꿍은 복수심에 불타 아예 늑대들을 양들이 있는 울타리안에 마구 집어넣어 양들을 잡아먹게 했다.

 

하지만 양들은 보통 양들이 아니다. 같은 울타리 영역 안의 양들의 숫자가 늑대의 숫자보다 더 많을 경우 늑대가 전부 잡아먹힌다. 물론 그 외의 경우는 양이 전부 잡아먹히겠지만 말이다.

꿍은 워낙 똑똑했기 때문에 이들의 결과는 이미 알고있다. 만약 빈 공간을 '.'(점)으로 나타내고 울타리를 '#', 늑대를 'v', 양을 'k'라고 나타낸다면 여러분은 몇 마리의 양과 늑대가 살아남을지 계산할 수 있겠는가?

단, 울타리로 막히지 않은 영역에는 양과 늑대가 없으며 양과 늑대는 대각선으로 이동할 수 없다.

 

 

 

---

• 풀이 :

BFS로 울타리로 막혀있는 각 공간을 탐색한 후 양과 늑대의 수를 카운트하여 양이나 늑대가 이기는 경우를 카운트해준다.

 

 

 

---

• 소스코드 : 

 

dx = [-1,1,0,0]
dy = [0,0,-1,1]
 
def bfs(x,y):
    queue = [[x,y]]
    global wolf
    global sheep
    sheepcnt = 0
    wolfcnt = 0
    if maps[x][y] == "v":
        wolfcnt +=1
    elif maps[x][y] == "k":
        sheepcnt+=1
    maps[x][y] = "#"
    while queue:
        ax = queue[0][0]
        ay = queue[0][1]
        del queue[0]
        for i in range(4):
            bx = ax + dx[i]
            by = ay + dy[i]
            if 0 <= bx < N and 0 <= by < M:
                if maps[bx][by] == ".":
                    queue.append([bx,by])
                elif maps[bx][by] == "v":
                    queue.append([bx,by])
                    wolfcnt +=1
                elif maps[bx][by] == "k":
                    queue.append([bx,by])
                    sheepcnt +=1
                maps[bx][by] = "#"
    if wolfcnt >= sheepcnt:
        wolf += wolfcnt
    else:
        sheep += sheepcnt
 
 
if __name__ == "__main__":
    N,M = map(int,input().split())
    maps = []
    for i in range(N):
        maps.append(list(input()))
    sheep = 0
    wolf = 0
    for i in range(N):
        for j in range(M):
            if maps[i][j] != "#":
                bfs(i,j)
    print(sheep,wolf)