youngyoungsunny / TIL

Today I Learned
1 stars 0 forks source link

BFS #3

Open youngyoungsunny opened 3 years ago

youngyoungsunny commented 3 years ago

최소거리 구하는 문제

youngyoungsunny commented 3 years ago

def bfs(graph, start, visited):

큐(Queue) 구현을 위해 deque 라이브러리 사용

queue = deque([start])
visited[start]=True #현재 노드를 방문 처리
while queue:
    v = queue.popleft() #큐에서 하나의 원소를 뽑아 출력하기
    print(v, end=' ')
    for i in graph[v]: #아직 방문하지 않은 인접한 원소들을 큐에 삽입
        if not visited[i]:
            queue.append(i)
            visited[i] = True #방문 처리

graph = [ [], [2,3,8], [1,7], [1,4,5] ]

visited = [False]*9
youngyoungsunny commented 3 years ago

Python - 데크(deque) 언제, 왜 사용해야 하는가?

-데크(deque)의 개념 : 보통 큐(queue)는 선입선출(FIFO) 방식으로 작동함. <---> 반면, 양방향 큐가 있는데 그것이 바로 데크(deque)!! 즉, 앞, 뒤 양쪽 방향에서 엘리먼트(element)를 추가하거나 제거 가능

-#데크(deque) 사용법
from collections import deque

deq = deque()

# Add element to the start
deq.appendleft(10)

# Add element to the end
deq.append(0)

# Pop element from the start
deq.popleft()

# Pop element from the end
deq.pop()

데크(deque)에 존재하는 메서드(method)는 대략 다음과 같다.

deque.append(item): item을 데크의 오른쪽 끝에 삽입한다. deque.appendleft(item): item을 데크의 왼쪽 끝에 삽입한다. deque.pop(): 데크의 오른쪽 끝 엘리먼트를 가져오는 동시에 데크에서 삭제한다. deque.popleft(): 데크의 왼쪽 끝 엘리먼트를 가져오는 동시에 데크에서 삭제한다. deque.extend(array): 주어진 배열(array)을 순환하면서 데크의 오른쪽에 추가한다. deque.extendleft(array): 주어진 배열(array)을 순환하면서 데크의 왼쪽에 추가한다. deque.remove(item): item을 데크에서 찾아 삭제한다. deque.rotate(num): 데크를 num만큼 회전한다(양수면 오른쪽, 음수면 왼쪽). 여기서 rotate() 메서드(method)가 특히 재밌는데, 설명이 부족하니 코드를 추가해 보겠다.

# Contain 1, 2, 3, 4, 5 in deq
deq = deque([1, 2, 3, 4, 5])

deq.rotate(1)
print(deq)
# deque([5, 1, 2, 3, 4])

deq.rotate(-1)
print(deq)
# deque([1, 2, 3, 4, 5])
#이렇게 양수 값 또는 음수 값을 파라미터(parameter)로 제공하여 데크(deque)를 좌, 우로 회전할 수 있다.

def solution(m, n, tomatoes): count = 0 # Count number of days deq = deque()

deq = list()

D = [(0, 1), (1, 0), (0, -1), (-1, 0)]

def search(row, col):
    searched_list = []

    for i, j in D:
        if (row + i < N and col + j < M) and (row + i >= 0 and col + j >= 0):
            if tomatoes[row + i][col + j] == 0:
                tomatoes[row + i][col + j] = 1
                searched_list.append((row + i, col + j))

    return searched_list

# Add all riped tomatoes
for r in range(N):
    for c in range(M):
        if tomatoes[r][c] == 1:
            deq.append((r, c))

# Search begin
while deq:
    for _ in range(len(deq)):
        r, c = deq.popleft()
        for tomato in search(r, c):
            deq.append(tomato)
    count += 1

# Check unriped tomato(es) after search
for r in range(N):
    for c in range(M):
        if tomatoes[r][c] == 0:
            return -1

return count - 1

if name == "main": M, N = map(int, input().split(" ")) tomatoes = [[int(n) for n in input().split(" ")] for _ in range(N)] print(solution(M, N, tomatoes))