문제
얼마전에 나는 불면증에 시달렸지... 천장이 뚫어져라 뜬 눈으로 밤을 지새우곤 했었지. 그러던 어느 날 내 친구 광민이에게 나의 불면증에 대해 말했더니 이렇게 말하더군. "양이라도 세봐!" 정말 도움이 안되는 친구라고 생각했었지. 그런데 막상 또 다시 잠을 청해보려고 침대에 눕고 보니 양을 세고 있더군... 그런데 양을 세다보니 이걸로 프로그램을 하나 짜볼 수 있겠단 생각이 들더군 후후후... 그렇게 나는 침대에서 일어나 컴퓨터 앞으로 향했지.
양을 # 으로 나타내고 . 으로 풀을 표현하는 거야. 서로 다른 # 두 개 이상이 붙어있다면 한 무리의 양들이 있는거지. 그래... 좋았어..! 이걸로 초원에서 풀을 뜯고 있는 양들을 그리드로 표현해 보는거야!
그렇게 나는 양들을 그리드로 표현하고 나니까 갑자기 졸렵기 시작했어. 하지만 난 너무 궁금했지. 내가 표현한 그 그리드 위에 몇 개의 양무리가 있었는지! 그래서 나는 동이 트기 전까지 이 프로그램을 작성하고 장렬히 전사했지. 다음날 내가 잠에서 깨어났을 때 내 모니터에는 몇 개의 양무리가 있었는지 출력되어 있었지.
입력
첫 번째 줄은 테스트 케이스의 수를 나타나는 T를 입력받는다.
이후 각 테스트 케이스의 첫 번째 줄에서는 H,W 를 입력받는다. H는 그리드의 높이이고, W는 그리드의 너비이다. 이후 그리드의 높이 H 에 걸쳐서 W개의 문자로 이루어진 문자열 하나를 입력받는다.
- 0 < T ≤ 100
- 0 < H, W ≤ 100
출력
각 테스트 케이스마다, 양의 몇 개의 무리로 이루어져 있었는지를 한 줄에 출력하면 된다.
#백준 11123 양 한마리... 양 두마리...
#양:#, 풀:.
import sys
input=sys.stdin.readline
sys.setrecursionlimit(100000)
t=int(input()) #테스트케이스의 수
#방향벡터
dx=[-1,1,0,0]
dy=[0,0,-1,1]
def dfs(x,y): #dfs 정의하기
visited[x][y]=True #현재노드 방문처리
for i in range(4):
nx=x+dx[i]
ny=y+dy[i]
if(0<=nx<h and 0<=ny<w): #리스트를 벗어나지 않고,
if(visited[nx][ny]==False): #아직 방문하지 않았고,
if(field[nx][ny]=='#'): #양이면
dfs(nx,ny)#재귀 돌기기
#케이스만큼 반복
for _ in range(t):
h,w=map(int, input().split()) #그리드의 높이, 너비
field=[]
for i in range(h):
field.append(list(input().strip()))
visited=[[False]*w for _ in range(h)] #행h,열w의 방문 리스트
cnt=0
for x in range(h):
for y in range(w):
if(visited[x][y]==False):
if(field[x][y]=='#'): #양 무리만 확인하기!
dfs(x,y)
cnt+=1 #한 양 무리 발견
print(cnt)
DFS 문제는 나만의 풀이 코드가 만들어진 거 같아서 한 두문제만 더 풀어보면 완전 쉽게 풀 수 있을 거 같다 희흐
'코딩 테스트 일지 📒' 카테고리의 다른 글
[다이나믹 프로그래밍] 1로 만들기 (0) | 2024.04.08 |
---|---|
[다이나믹 프로그래밍] list를 value로 갖는 dict 생성하기, enumerate() 함수, dict 첫 번째 key, value 가져오기 + 개미전사 (1) | 2024.04.07 |
[그리디] 백준_흙길 보수하기 (0) | 2024.04.01 |
[구현] 백준 5430_AC (0) | 2024.04.01 |
[DFS] 백준_양치기 꿍 (1) | 2024.04.01 |