[Python] 백준 #2667 단지번호붙이기

www.acmicpc.net/problem/2667

2667번: 단지번호붙이기

dfs로 해결

점점 감을 잡고있다.!

 

def dfs(x,y):
    global cnt
    if x<0 or y<0 or x>=n or y>=n:
        return False
    
    if l_list[x][y]==1:
        l_list[x][y]=0
        cnt+=1
        
        dfs(x-1,y) #상
        
        dfs(x+1,y) #하
        
        dfs(x,y-1) #좌
        
        dfs(x,y+1) #우
        
        return (int(1),cnt)
    
n=int(input())
l_list=[]
result=[] # 단지내 집의 수
ptr=0
cnt=0

for i in range(n):
    l_list.append(list(map(int,input())))
    
for i in range(n):
    for j in range(n):        
        a=dfs(i,j)        
        if(a!=None):    
            ptr+=1            
            result.append(a[1])
            cnt=0
print(ptr)
result=sorted(result)
for i in result:
    print(i)