백준 / dfs와 bfs / 1260

Question

문제링크
Silver 2

Logic

기본 구조 : 직접구현, dfs, bfs
1. 우선 주어진 그래프로 2차원 배열을 만든다. 이는 간선의 유무를 판단한다.
2. 주어진 그래프를 dfs 방식과 bfs 방식으로 출력하는 구문을 작성한다.
3. def는 재귀 방식을 취한다. 우선 본인을 출력한다. 다음 방문하지 않았고, 간선이 존재하는 자식에 대해 재귀호출 한다.
4. bfs는 큐를 활용한다.

Code

from sys import stdin

N,M,V = map(int,stdin.readline().split())
visit = [0 for i in range(N+1)]
li = [[0 for __ in range(N+1)] for _ in range(N+1)]

for i in range(M):
    a,b = map(int,stdin.readline().split())
    li[a][b]=1
    li[b][a]=1

def dfs(v):
    print(v,end=' ')
    visit[v]=1
    for i in range(1,N+1):
        if visit[i]==0 and li[v][i]==1:
            dfs(i)

def bfs(v):
    queue = [v]
    visit[v] = 0
    while(queue):
        v=queue[0]
        print(v, end=' ')
        del queue[0]
        for i in range(1,N+1):
            if visit[i]==1 and li[v][i]==1:
                queue.append(i)
                visit[i]=0

def li_print(li):
    for s in li:
        print(*s)

dfs(V)
print()
bfs(V)

li_print(li)