백준 python 기록

백준 1260 python DFS와 BFS

우히힝 2021. 7. 9. 23:24

백준 링크 : https://www.acmicpc.net/problem/1260

 

1260번: DFS와 BFS

첫째 줄에 정점의 개수 N(1 ≤ N ≤ 1,000), 간선의 개수 M(1 ≤ M ≤ 10,000), 탐색을 시작할 정점의 번호 V가 주어진다. 다음 M개의 줄에는 간선이 연결하는 두 정점의 번호가 주어진다. 어떤 두 정점 사

www.acmicpc.net

DFS와 BFS의 기본적인 문제였다.

이건 DFS와 BFS에 대한 기본적인 지식을 습득하고 도전해야한다.

DFS는 깊이우선탐색. 동굴을 넓히듯이 탐색하는 느낌이다.

BFS는 부채꼴로 펼치듯이 탐색하는 느낌.

나는 구글에서 검색해서 따로 공부했었다.

https://devuna.tistory.com/32 이런 사이트 등등 매우 자세히 설명해주신 분들이 많다.

문제시 삭제할게용..

N,M,V=map(int,input().split())
matrix=[[0]*(N+1) for i in range(N+1)]
for i in range(M):
    a,b = map(int,input().split())
    matrix[a][b]=matrix[b][a]=1
visit_list=[0]*(N+1)

def dfs(v):
    visit_list[v] = 1
    print(v,end=" ")
    for i in range(1,N+1):
        if visit_list[i] == 0 and matrix[v][i] == 1:
            dfs(i)
def bfs(v):
    queue=[v]
    visit_list[v]=0
    while queue:
        v=queue.pop(0)
        print(v,end=" ")
        for i in range(1,N+1):
            if visit_list[i] == 1 and matrix[v][i] == 1:
                queue.append(i)
                visit_list[i] = 0

dfs(V)
print()
bfs(V)