[Gold V] Title: 적록색약, Time: 140 ms, Memory: 115296 KB -BaekjoonHub

Author: wlsgur11

백준/Gold/10026. 적록색약/README.md

@@ -4,15 +4,15 @@
 
 ### 성능 요약
 
-메모리: 115152 KB, 시간: 148 ms
+메모리: 115296 KB, 시간: 140 ms
 
 ### 분류
 
 너비 우선 탐색, 깊이 우선 탐색, 그래프 이론, 그래프 탐색
 
 ### 제출 일자
 
-2025년 3월 29일 12:08:57
+2025년 3월 29일 12:12:33
 
 ### 문제 설명
 

백준/Gold/10026. 적록색약/적록색약.py

@@ -1,7 +1,7 @@
 from collections import deque
 
 
-def bfs(i, j, A):
+def bfs(i, j, A, visited):
     q = deque([(i, j)])
 
     while q:
@@ -17,32 +17,29 @@ def bfs(i, j, A):
 change_normal = {'R': 1, 'G': 2, 'B': 3}
 change_blind = {'R': 1, 'G': 1, 'B': 2}
 
-first = [list(map(str, input())) for _ in range(N)]
+RGB = [list(map(str, input())) for _ in range(N)]
 
 normal = [[0] * N for _ in range(N)]
 blind = [[0] * N for _ in range(N)]
 for i in range(N):
     for j in range(N):
-        normal[i][j] = change_normal[first[i][j]]
-        blind[i][j] = change_blind[first[i][j]]
+        normal[i][j] = change_normal[RGB[i][j]]
+        blind[i][j] = change_blind[RGB[i][j]]
 
-visited = [[False] * N for _ in range(N)]
+visited1 = [[False] * N for _ in range(N)]
+visited2 = [[False] * N for _ in range(N)]
 dx = [-1, 1, 0, 0]
 dy = [0, 0, -1, 1]
 
 ans = [0, 0]
 
 for i in range(N):
     for j in range(N):
-        if not visited[i][j]:
-            bfs(i, j, normal)
+        if not visited1[i][j]:
+            bfs(i, j, normal, visited1)
             ans[0] += 1
-
-visited = [[False] * N for _ in range(N)]
-for i in range(N):
-    for j in range(N):
-        if not visited[i][j]:
-            bfs(i, j, blind)
+        if not visited2[i][j]:
+            bfs(i, j, blind, visited2)
             ans[1] += 1
 
 print(*ans)