글 읽기 - [파이썬] 정답이긴 한데 시간을 어디서 줄일 수 있을까요??

bnk159hair 2년 전 0

학과에서 C++만 사용하다가 파이썬을 사용해보니 문법이 달라서 이해가 안되는 것들도 있고

알고리즘 공부를 시작한지 얼마 안되었습니다.

그래서 이렇게 질문을 올립니다.

제 코드는 시간이 600ms정도 소요되는 거 같습니다.

그런데 다른 분들 코드는 100ms 아래로 나오게 잘 짜셨더라구요.

코드의 알고리즘은 간단한데 어느 부분에서 이렇게 차이가 많이 나는지 궁금합니다.

제 코드에서 미숙한 부분이나 최적화할 수 있는 부분들이 무엇이 있을까요?

from queue import Queue 
from collections import defaultdict

num_of_V, num_of_L, Start = map(int,input().split())

graph = defaultdict(list)
for _ in range(num_of_L):
    A, B =map(int, input().split())
    graph[A].append(B)
    graph[B].append(A)

for i in graph:
    graph[i].sort()
    
def DFS(graph, start_node):
    visit = {}
    stack = []
    
    stack.append(start_node)
    
    while stack:
        node = stack.pop()
        if node not in visit:
            visit[node] =True
            stack.extend(graph[node])
    return list(visit.keys())

def BFS(graph, start_node):
    visit = {}
    queue = Queue()
    
    queue.put(start_node)
    
    while queue.empty()==False:
        node = queue.get()
        if node not in visit:
            visit[node]=True
            for adNode in graph[node]:
                queue.put(adNode)
    return list(visit.keys())

temp = BFS(graph, Start)
for i in graph:
    graph[i].reverse()

print(" ".join(str(i) for i in DFS(graph, Start)))

print(" ".join(str(i) for i in temp))

ai4youej 2년 전 1

빠른 입출력을 사용하면 충분히 100ms대 나올 거 같습니다!

bnk159hair 2년 전 1

감사합니다!! 시간이 거의 1/3 수준으로 줄어들었어요 ㅎㅎ

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

문제

문제

출처

ICPC

[파이썬] 정답이긴 한데 시간을 어디서 줄일 수 있을까요??

bnk159hair 2년 전 0

ai4youej 2년 전 1

bnk159hair 2년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말