백준 2060 바이러스 DFS

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

2606번: 바이러스

 

• 연결리스트를 만든다
• 방문 배열을 만든다
• DFS를 구현해서 방문배열을 업데이트 시켜준다
• 방문배열에서 True인 갯수를 센다.
  • 나는 for문을 돌려서 셌는데 
  • 다른 코드 찾아보니 sum 함수 이용해서 하면 한 줄로 간단하게 풀 수 있더라

 

N = int(input())
M = int(input())

graph = [[] for _ in range(N+1)]
visited = [False] * (N+1)

for _ in range(M):
    s, e = map(int, input().split())
    graph[s].append(e)
    graph[e].append(s)

answer = -1

def dfs(x):
    if visited[x]:
        return
    
    visited[x] = True
    
    for i in graph[x]:
        if visited[i] != True:
            dfs(i)
dfs(1)

'''
for i in range(1,len(graph)):
    if visited[i] == True:
        answer += 1
print(answer)
'''

print(sum(visited)-1)