[백준(BOJ)] 10026번 : 적록색약 - PYTHON[파이썬]

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

10026번: 적록색약

 

---

⊙ 문제

⊙ 입력

⊙ 출력

⊙ 예제 입출력

⊙ 알고리즘 분류

• 그래프 이론
• 그래프 탐색
• 너비 우선 탐색
• 깊이 우선 탐색

 

---

 

⊙ 문제 접근 과정

 

bfs를 돌렸다. 2번.

 

한 번은 입력받은 RGB 그대로 돌려서 구역을 찾아냈다.

 

마지막 한번은 R을 전부 G로 바꾸고 visited 여부를 초기화해준 후 돌려줬다.

 

---

 

⊙ 문제 풀이

 

import sys
from collections import deque

input = sys.stdin.readline
N = int(input())
arr = [list(map(str, input().strip())) for _ in range(N)]
visited = [[0]*(N+1) for _ in range(N+1)]
dx = [0,0,-1,1]
dy = [1,-1,0,0]

q = deque()
ans = 0

def bfs(x,y):
    global ans
    q.appendleft([x,y])
    visited[x][y]=1
    while q:
        x,y = q.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if 0<=nx<N and 0 <=ny<N and visited[nx][ny]==0:
                if arr[x][y]==arr[nx][ny]:
                    visited[nx][ny]=1
                    q.append([nx,ny])
        
    
for i in range(N):
    for j in range(N):
        if visited[i][j]==0:
            bfs(i,j)
            ans+=1
print(ans,end=' ')
for i in range(N):
    for j in range(N):
        if arr[i][j]=='R':
            arr[i][j]='G'
visited = [[0]*(N+1) for _ in range(N+1)]
ans = 0
for i in range(N):
    for j in range(N):
        if visited[i][j]==0:
            bfs(i,j)
            ans+=1
print(ans)

 

---

⊙ 결과

 

---

⊙ 마무리

 

 

NONE

 

 

좋아요는 로그인하지 않아도 누를 수 있습니다!