20210806#(73) 백준 1260번 DFS와 BFS

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

1260번: DFS와 BFS

 

제출 코드--------------------------------------------------------------------------------------------------------------------------------------

 

def dfs(v):
    print(v, end=' ')
    visit[v]=1
    for i in range(1, n+1):
        if visit[i]==0 and graph[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 graph[v][i]==1:
                queue.append(i)
                visit[i]=0
            
n,m,v=map(int,input().split())

graph =[[0]*(n+1) for _ in range(n+1)]
visit=[0]*(n+1)

for i in range(m):
    x,y=map(int,input().split())
    graph[x][y]=1
    graph[y][x]=1
    
dfs(v)
print()
bfs(v)

 

 

참고코드---------------------------------------------------------------------------------------------------------------------------------------

#1 

def dfs(v):
    print(v, end=' ')
    visit[v] = 1
    for i in range(1, n + 1):
        if visit[i] == 0 and s[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 s[v][i] == 1:
                queue.append(i)
                visit[i] = 0

n, m, v = map(int, input().split())
s = [[0] * (n + 1) for i in range(n + 1)]
visit = [0 for i in range(n + 1)]
for i in range(m):
    x, y = map(int, input().split())
    s[x][y] = 1
    s[y][x] = 1
    
dfs(v)
print()
bfs(v)

 

 

 

#2

from collections import deque
import sys
read = sys.stdin.readline

def bfs(v):
  q = deque()
  q.append(v)       
  visit_list[v] = 1   
  while q:
    v = q.popleft()
    print(v, end = " ")
    for i in range(1, n + 1):
      if visit_list[i] == 0 and graph[v][i] == 1:
        q.append(i)
        visit_list[i] = 1

def dfs(v):
  visit_list2[v] = 1        
  print(v, end = " ")
  for i in range(1, n + 1):
    if visit_list2[i] == 0 and graph[v][i] == 1:
      dfs(i)

n, m, v = map(int, read().split())

graph = [[0] * (n + 1) for _ in range(n + 1)] 
visit_list = [0] * (n + 1)
visit_list2 = [0] * (n + 1)

for _ in range(m):
  a, b = map(int, read().split())
  graph[a][b] = graph[b][a] = 1

dfs(v)
print()
bfs(v)