1260번 / DFS와 BFS / 파이썬

📕 문제

https://velog.io/@falling_star3/백준Python-1260번-DFS와BFS

[백준][Python] 1260번 DFS와 BFS(DFS/BFS 기본 구현 자세히)

 

📗실행결과

 

 

📄 내가 생각한 풀이

 

💻 답안

 #1260

n,m,v =map(int, input().split())
# n : 정점의 개수, m : 간선 연결하는 두 정점의 번호. v : 탐색 시작할 정점 번호


 # 방문한 리스트 모음, 초기값 0
visited1 = [0]*(n+1)
visited2 = visited1.copy()


# 0으로 채워진 n개의 2차원 행렬 생성
graph = [[0]*(n+1) for _ in range(n+1)]

# 두 정점의 번호를 그래프상 1로 변경
for i in range (m):
    a,b = map(int, input().split())
    graph[a][b] = graph[b][a] = 1



# dfs 함수 - 재귀함수
def dfs(v):
   
    # 방문시 1로 변경
    visited1[v] = 1
    print(v, end = ' ')

    for i in range(1,n+1):
        if graph[v][i] == 1 and visited1[i] == 0:
            dfs(i)

# bfs함수
def bfs(v):
  

    queue = [v]
    visited2[v] = 1    
    while queue:
        v = queue.pop(0)
        print(v, end =' ')
        for i in range(1,n+1):
            if(visited2[i] == 0 and graph[v][i] == 1):
                queue.append(i)
                visited2[i] = 1
dfs(v)
print()
bfs(v)

 

☺️ 내 코드에서 보완해야 할 점

어려워서 참고 자료보고 거의 해결. 알고리즘 공부 더 해야함

 

😉 참고

https://velog.io/@falling_star3/백준Python-1260번-DFS와BFS

[백준][Python] 1260번 DFS와 BFS(DFS/BFS 기본 구현 자세히)