글 읽기 - 맞았는데 코드 개선 사항이 있을까 하여 질문드립니다

david02324 3년 전 0

bfs와 dfs모두를 사용해서 작성해봤습니다. 둘 다 정답처리는 됐는데, bfs는 나름 만족하지만, dfs부분이 가독성도 떨어지고 좀 지저분해 보입니다. 혹시 고수분들 중에서 제 dfs코드의 개선점이나 조언등을 해주실 수 있나요..?

from sys import stdin
from collections import deque

input = stdin.readline

n,m = map(int,input().split())

# 정점 저장
nodes = [[] for _ in range(n+1)]
# 방문 처리
visited = [False] * (n+1)
for _ in range(m):
    i,j = map(int,input().split())
    nodes[i].append(j)
    nodes[j].append(i)

# 너비 우선 탐색
def bfs(start):
    global nodes,visited
    if visited[start]:
        return 0

    visited[start] = True
    queue = deque([start])

    while queue:
        for node in nodes[queue.popleft()]:
            if not visited[node]:
                visited[node] = True
                queue.append(node)

    return 1

# 깊이 우선 탐색
def dfs(start):
    global nodes,visited
    if visited[start]:
        return 0
    
    visited[start] = True
    stack = [start]

    if not nodes[stack[-1]]:
        return 1
    while stack:
        for idx,node in enumerate(nodes[stack[-1]]):
            if not visited[node]:
                visited[node] = True
                stack.append(node)
                break
            if idx == len(nodes[stack[-1]]) - 1:
                stack.pop()
    
    return 1

count = 0
for i in range(1,n+1):
    count += dfs(i)
print(count)

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

맞았는데 코드 개선 사항이 있을까 하여 질문드립니다

david02324 3년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말