백준 (boj) 파이썬 - 10026 : 적록색약

#문제링크 : https://www.acmicpc.net/problem/10026

10026번: 적록색약

 

# 분류 : dfs

 

# 난이도 : Gold 5

 

 

적록색약인 경우, 적록색약이 아닌 경우를 나누어 총 2번 dfs를 돌렸다. 시간 초과가 우려되었으나 입력이 100줄 이하라서 문제없이 통과 되었다. 

 

#코드(Python)

from sys import setrecursionlimit

setrecursionlimit(10000)
n = int(input())
grid = []
visited = [[False for i in range(n)] for i in range(n)]

cnt = 0
for _ in range(n):
    grid.append(list(input()))

color = ""
colors = []
colors2 = []


def is_valid(x,y):
    if x >= 0 and x < n and y >= 0 and y < n:
        return True
    return False

def dfs(g,x,y):
    global color 
    if not is_valid(x,y):
        return False
    if visited[x][y] == False and g[x][y] == color:
        visited[x][y] = True
        global cnt
        cnt += 1
        dfs(g,x-1,y)
        dfs(g,x+1,y)
        dfs(g,x,y+1)
        dfs(g,x,y-1)
        return True
    return False

for x in range(n):
    for y in range(n):
        if grid[x][y] != 0:
            color = grid[x][y]
            if dfs(grid,x,y):
                colors.append(color)

visited = [[False for i in range(n)] for i in range(n)]

for x in range(n):
    for y in range(n):
        if grid[x][y] == "R":
            grid[x][y] = "G"

for x in range(n):
    for y in range(n):
        if grid[x][y] != 0:
            color = grid[x][y]
            if dfs(grid,x,y):
                colors2.append(color)

print(len(colors) , len(colors2))