🔍 Problem

10026 적록색약

---

⚡️ 사용한 알고리즘

BFS

---

📃 Input&Output

---

📒 해결 과정

1️⃣ 먼저 적록색약 BFS와 적록색약이 아닌 BFS를 나눈다.
2️⃣ 적록색약 매트릭스에서 R과 G를 같은 것으로 처리하므로, G를 R로 바꿔준다.
3️⃣ 각각 BFS 진행하고 count를 구한다.

---

❗주의점

1. 파이썬에는 immutable, mutable이 있고 immutable은 가변 불가, mutable은 가변가능한 클래스들이다. 따라서 이 문제는 하나의 bfs함수를 사용하고 있지만, count 를 따로 사용하고 있기때문에 count를 리스트의 인덱스값으로 설정하여 구하였다.

   immutable : 정수, 실수, 문자열, 튜플
   mutable : 리스트, 딕셔너리

---

💻 Code

import sys
from collections import deque
import copy

N = int(sys.stdin.readline())

mat = [list(sys.stdin.readline().strip()) for _ in range(N)]
vst = [[False] * N for _ in range(N)]

dx = [-1, 1, 0, 0]
dy = [0, 0, -1, 1]

normal_cnt=0
anomal_cnt=0


def bfs(i, j,vst,mat,cnt,k):
    que = deque([])
    que.append((i, j))
    vst[i][j] = True

    while que:
        x, y = que.popleft()
        for i in range(4):
            nx, ny = x + dx[i], y + dy[i]
            if nx >= 0 and ny >= 0 and nx < N and ny < N:
                if mat[nx][ny] == mat[x][y] and vst[nx][ny] == False:
                    que.append((nx, ny))
                    vst[nx][ny] = True
                    
    if k=='x':
        cnt[0]+=1
    else:
        cnt[1]+=1
    

# 적록색약 X

cnt=[0,0]

for i in range(N):
    for j in range(N):
        if vst[i][j] == False:
            bfs(i, j,vst,mat, cnt,'x')

rg_mat=copy.deepcopy(mat)
for i in range(N):
    for j in range(N):
        if rg_mat[i][j]== 'G':
            rg_mat[i][j] = 'R'
            


# vst 초기화
for i in range(N):
    for j in range(N):
        vst[i][j] = False

        

# 적록색약 O
for i in range(N):
    for j in range(N):
        if vst[i][j] == False:
            bfs(i, j,vst,rg_mat, cnt,'o')

print(f"{cnt[0]} {cnt[1]}")

---

🤔 느낀점

실제 네이버 같은 코테에서는 디버깅 못할텐데, 코드 커지면 걱정된다.