더북(TheBook)

코드 6-6

from queue import Queue

class Graph:
    def __init__(self, vertex_num):
        # 인접 리스트로 구현
        self.adj_list = [[] for _ in range(vertex_num)]
        # 방문 여부 체크
        self.visited = [False for _ in range(vertex_num)]

    def add_edge(self, u, v):
        self.adj_list[u].append(v)
        self.adj_list[v].append(u)

    def init_visited(self):
        for i in range(len(self.visited)):
            self.visited[i] = False

코드 6- 6을 보면 queue 모듈에서 Queue 클래스를 가져옵니다. 파이썬이 제공하는 큐에는 enqueue 연산을 하는 put() 메서드와 dequeue 연산을 하는 get() 메서드가 있습니다.

신간 소식 구독하기
뉴스레터에 가입하시고 이메일로 신간 소식을 받아 보세요.