Merge pull request #36 from YoonYn9915/main

YoonYn9915 / 6월 5주차 / 2문제

Author: Mingguriguri

YoonYn9915/Graph/2024-07-01-[백준]-#10026-적록색약.py

@@ -0,0 +1,68 @@
+from collections import deque
+
+# 2차원 배열을 이용한 bfs 구현
+def bfs(x, y):
+
+    # 2차원 배열에서 현재 노드의 상하좌우를 검사하기 위한 dx와 dy
+    dx = [0, -1, 0, 1]
+    dy = [-1, 0, 1, 0]
+
+    # queue를 이용한 bfs 그래프 탐색
+    queue.append((x, y))
+    visited[x][y] = 1
+
+
+    while queue:
+        (x, y) = queue.popleft()
+
+        # queue에서 pop한 현재 노드를 기준으로 상하좌우 탐색
+        for k in range(4):
+            row = x + dx[k]
+            col = y + dy[k]
+
+            # 탐색 조건
+            # 1. 현재 노드가 배열의 인덱스 범위 안인지 0 <= x < n and 0 <= y < n
+            # 2. 같은 색인지
+            # 3. 방문하지 않았는지
+            if 0 <= row < n and 0 <= col < n:
+                if arr[x][y] == arr[row][col] and visited[row][col] == 0:
+                    queue.append((row, col))
+                    visited[row][col] = 1
+
+
+# 초기값 세팅
+n = int(input())
+visited = [[0] * n for _ in range(n)]
+arr = [list(input()) for _ in range(n)]
+queue = deque()
+
+# 적록색약이 아닌 경우의 답
+answerForNormal = 0
+
+# 적록색약인 경우의 답
+answerForColorBlindness = 0
+
+
+# 적록색약이 아닌 경우
+for i in range(n):
+    for j in range(n):
+        if visited[i][j] == 0:
+            bfs(i, j)
+            answerForNormal += 1
+
+
+# 적록색약인 경우 R과 G는 같으므로
+
+for i in range(n):
+    for j in range(n):
+        if arr[i][j] == 'G':
+            arr[i][j] = 'R'
+
+visited = [[0] * n for _ in range(n)]
+for i in range(n):
+    for j in range(n):
+        if not visited[i][j]:
+            bfs(i,j)
+            answerForColorBlindness += 1
+
+print(answerForNormal, answerForColorBlindness)

YoonYn9915/Graph/2024-07-01-[백준]-#4963-섬의 개수.py

@@ -0,0 +1,54 @@
+from collections import deque
+
+# 지도(2차원 배열)에서 섬을 탐색하기 위한 bfs
+def bfs(x, y):
+
+    # 탐색범위가 현재 노드 기준 가로, 세로, 대각선이므로 총 8가지
+    dx = [-1, -1, -1, 0, 0, 1, 1, 1]
+    dy = [-1, 0, 1, -1, 1, -1, 0, 1]
+
+    queue.append((x, y))
+    visited[x][y] = 1
+
+    # queue가 존재하면
+    while queue:
+
+        # queue에서 pop해서
+        x, y = queue.popleft()
+
+        #가로, 세로 대각선에 있는 노드가 다음 세가지 조건을 만족하는지 검사
+        # 탐색 조건
+        # 1. 현재 노드가 배열의 인덱스 범위 안인지 0 <= row < h and 0 <= col < w
+        # 2. 땅인지 (arr값이 1인지)
+        # 3. 방문하지 않았는지
+        for k in range(8):
+            row = x + dx[k]
+            col = y + dy[k]
+
+            if 0 <= row < h and 0 <= col < w and arr[row][col] == 1 and visited[row][col] == 0:
+                queue.append((row, col))
+                visited[row][col] = 1
+
+
+answer = deque()
+queue = deque()
+
+while True:
+    w, h = map(int, input().split())
+    if h == 0 and w == 0:
+        break
+
+    visited = [[0] * w for _ in range(h)]
+    arr = [list(map(int, input().split())) for _ in range(h)]
+    ans = 0
+
+    for i in range(h):
+        for j in range(w):
+            if arr[i][j] == 1 and visited[i][j] == 0:
+                bfs(i, j)
+                ans += 1
+
+    answer.append(ans)
+
+for ans in answer:
+    print(ans)