RGB로 구분되는 n 크기의 이차원 배열을 입력받는다. 그 후 일반인이 느끼는 구역과 적록색약이 느끼는 구역을 출력하는 문제. 입력받는 이차원 배열을 일반인과 색약으로 구분하여 저장하고 각각 방문확인 배열을 만들어 dfs 연산을 이용하여 풀었다.
\nimport sys\ninput = sys.stdin.readline\n\ndire = [[1, 0],\n [-1, 0],\n [0, 1],\n [0, -1]]\n\n\ndef dfs(sy, sx, color, tgraph, tvisit):\n q = [[sy, sx]]\n tvisit[sy][sx] = True\n\n while q:\n y, x = q.pop()\n for ty, tx in dire:\n ny, nx = y + ty, x + tx\n if 0 <= ny < n and 0 <= nx < n:\n if not tvisit[ny][nx] and tgraph[ny][nx] == color:\n q.append([ny, nx])\n tvisit[ny][nx] = True\n\nn = int(input())\ngraph = [[] for _ in range(n)]\ncolorweekgraph = [[] for _ in range(n)]\n\nfor i in range(n):\n t = input()\n graph[i] = list(t)\n colorweekgraph[i] = list(map(lambda x: 'R' if x == 'G' else x, t))\n\nvisit = [[False for _ in range(n)] for _ in range(n)]\ncolorweekvisit = [[False for _ in range(n)] for _ in range(n)]\n\narea, colorweekarea = 0, 0\nfor y in range(n):\n for x in range(n):\n if not visit[y][x]:\n dfs(y, x, graph[y][x], graph, visit)\n area += 1\n if not colorweekvisit[y][x]:\n dfs(y, x, colorweekgraph[y][x], colorweekgraph, colorweekvisit)\n colorweekarea +=1\n\nprint(area, colorweekarea)