2468번: 안전 영역 (acmicpc.net)

from collections import deque
n = int(input())

graph = []

for _ in range(n):
    graph.append(list(map(int, input().split())))

max_h = 0
for i in range(n):
    if max(graph[i])>max_h:
        max_h = max(graph[i])

dx = [1, -1, 0, 0]
dy = [0, 0, 1, -1]

def bfs(x, y, k):
    q = deque([(x,y)])
    while q:
        (x, y) = q.popleft()
        for i in range(4):
            nx = x+dx[i]
            ny = y+dy[i]
            if nx>=n or nx<0 or ny>=n or ny<0:
                continue
            if graph[ny][nx] >k and visited[ny][nx]==0:
                q.append((nx, ny))
                visited[ny][nx] = 1

max_count = 0
for x in range(max_h+1):
    res = 0
    visited = [[0]*n for i in range(n)]
    for i in range(n):
        for j in range(n):
            if graph[j][i]>x and visited[j][i]==0:
                bfs(i,j,x)
                res+=1
    if res > max_count:
        max_count = res

print(max_count)

#시간 복잡도 문제

음..

# visited 리스트를 안 만들면 안 되는 이유:
그걸 안 만들면 한번 방문했던 좌표를 -1처럼 엄청 작은 수로 설정해서 k보다 확실히 작아지도록 하려고 했지만..

k값을 조정하면서 몇 개의 덩어리가 나오는지를 구하는 전체 과정을 계속 반복해야 하기 때문에
처음 input 한 그래프는 계속 그 값을 유지하고 있어야 한다. 
반복할 때마다 그래프를 새로 만드는 것보다 
visited 그래프를 초기화하는 것이 100배 쉬울 것 같다. 

# 주의할 점: 

이차원 그래프에서 graph[y][x]이렇게 해야 x값이 가로, y값이 세로가 된다는 것..이 헷갈린다.

DFS란?

Depth-First Search의 약자로, 깊이 우선 탐색 방법.

• 한 방향으로 갈 수 있을 때까지 계속 가다가 더이상 갈 수 없게 되면 가장 가까운 갈림길로 돌아와서(스택에서 하나씩 pop하면서 인접노드 중 방문하지 않은 게 나올 때까지 탐색) 다시 깊숙히 들어가는 방향으로 진행한다.

#dfs 예제

def dfs(graph, v, visited):
    visited[v] = True
    print(v, end=' ')

    for i in graph[v]:
        if not visited[i]:
            dfs(graph, i, visited)

#각 노드가 연결된 정보는 2차원 리스트로 표현
graph = [
    [],
    [2, 3, 8],
    [1, 7],
    [1, 4, 5], 
    [3, 5],
    [3, 4],
    [7],
    [2, 6, 8],
    [1, 7]
]

#각 노드가 방문된 정보를 리스트 자료형으로 표현(1차원 리스트)
visited = [False] * 9

dfs(graph, 1, visited)

# 이것이 코딩테스트다 - with 파이썬에서 얼음 얼리기 문제 확인

--->

자료구조 중 스택을 이용한다.

자기자신을 계속 호출한다. 함수 자체 반복이 아니라 재귀함수를 여러번 반복해서 돌리면서 구현함.

재귀함수가 더이상 안 돌아가면 메인 코드의 반복문에서 값이 하나 올라가서 다음 탐색을 진행함. 

BFS란?

Breadth-First Search의 약자로, 넓이 우선 탐색 방법

• 임의의 노드에서 시작해서 인접한 노드-->멀리 떨어진 노드 순으로 탐색하는데, 해당 노드의 인접 노드 중 방문하지 않은 모든 노드를 큐에 삽입한다. 그다음 큐에 있는 노드를 하나씩 꺼내면서 인접노드를 방문처리하고 큐에 삽입하는 것을 반복한다. 인접 노드가 없으면 꺼내기만 한다. 남아 있는 노드에 방문하지 않은 인접노드가 없을 때까지 반복. 
• BFS는 재귀함수를 호출하지 않는다. 대신 방문한 노드들을 저장하고 차례대로 꺼낼 수 있는 Queue(Dequeue)를 사용한다.
• 두 노드 사이의 최단 경로를 찾고 싶을 때 이 방법으로 탐색한다. 

#bfs 예제

from collections import deque

def bfs(graph, start, visited):
    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

#각 노드가 연결된 정보는 2차원 리스트로 표현
graph = [
    [],
    [2, 3, 8],
    [1, 7],
    [1, 4, 5], 
    [3, 5],
    [3, 4],
    [7],
    [2, 6, 8],
    [1, 7]
]

#각 노드가 방문된 정보를 리스트 자료형으로 표현(1차원 리스트)
visited = [False] * 9

bfs(graph, 1, visited)

# 이것이 코딩테스트다 - with 파이썬에서 미로 탈출 문제 확인

---> Dequeue 사용, 함수 자체를 다 반복시키고 메인 코드에서는 그 bfs함수의 리턴값을 출력만 함

DFS는 스택

BFS는 큐

둘 다 시간 복잡도는 O(N)이지만 일반적으로 BFS가 DFS보다 좀 더 빠르다.