Python3
문제
https://www.acmicpc.net/problem/1012
키워드
- 그래프
- Connected Component
문제 풀이
문제 요구사항
- 한 배추의 상하좌우 네 방향에 다른 배추가 위치한 경우에 서로 인접해있다.
- 배추들이 모여있는 곳에는 배추흰지렁이가 한 마리만 있으면 되므로 서로 인접해있는 배추들이 몇 군데에 퍼져있는지 조사하면 총 몇 마리의 지렁이가 필요한지 알 수 있다.
=> Connected Component - 0은 배추가 심어져 있지 않은 땅이고, 1은 배추가 심어져 있는 땅을 나타낸다.
변수 및 함수 설명
- t : 테스트 케이스의 개수
- n : 배추를 심은 배추밭의 세로 길이 (1 ≤ N ≤ 50)
- m : 가로 길이 (1 ≤ M ≤ 50)
- k : 배추가 심어져 있는 위치의 개수 K(1 ≤ K ≤ 2500)
- x : 배추의 위치 X(0 ≤ X ≤ M-1)
- y : 배추의 위치 Y(0 ≤ Y ≤ N-1)
- adjMatrix : 배추의 위치가 들어 있는 인접행렬 (배추밭)
- visited : 방문했는지 확인하는 2차원 배열 (타입은 bool)
- bfsQueue : bfs에서 활용할 큐
- currentLocation : 큐에서 가장 첫 번째 있는 원소
- curY : 현재 y좌표
- curX : 현재 x좌표
- ny : 다음에 이동할 y좌표
- nx : 다음에 이동할 x좌표
- cnt : 필요한 지렁이 개수
풀이
BFS 사용
인접행렬 adjMatrix
(bfs 함수)
- 큐에 아무것도 남아 있지 않을 때까지 반복한다.
- 큐에 들어 있는 첫 번째 튜플을 꺼내 현재 좌표를 나타내는 curY, curX에 저장한다.
- 총 4번 반복한다. (상하좌우)
- 다음 이동할 칸은 현재 좌표에 dy[i], dx[i]를 더한 값이다. (시계 방향으로 이동함)
- 이때 다음 이동할 칸이 아래 조건 중 하나라도 만족하면 continue로 건너뛴다.
1) 현재 좌표가 (0, 0), (n-1, m-1) 일 때 다음 이동하는 곳이 밖이면 안 된다.
2) 다음 이동할 칸이 0이라면 갈 수 없다.
3) 이미 방문한 곳은 또 가지 않는다. - 모두 해당하지 않으면 방문 처리를 해주고, queue에 다음 이동하는 칸 튜플을 넣는다.
(입력 및 선언)
- 테스트 케이스 입력 받기
- 테스트 케이스 개수만큼 가로, 세로, 배추 위치를 입력 받는다.
- 배추 위치가 입력 되면 해당 위치의 adjMatrix에 1을 넣는다. (배추밭 만드는 과정)
- 방문 처리 배열도 2차원으로 선언하고 모두 False로 초기화시킨다.
- 지렁이 개수를 세는 cnt 변수도 여기에 넣어서 테스트케이스가 새로 시작할 때마다 0으로 초기화될 수 있도록 한다.
(배추 위치 찾기)
- 밭 전체를 돌기 위해 이중 반복문을 사용한다.
- 만약 배추밭(adjMatrix)가 1이고, 방문 처리가 안 된 곳이라면
- bfsQueue에 해당 좌표를 넣고, 방문 처리를 해준다.
- bfs 함수를 호출하고 지렁이 개수(cnt)를 하나 늘린다.
(최종 출력)
- 지렁이 개수(cnt)를 출력한다.
DFS 사용
Recursion 에러가 나서 재귀 깊이를 설정해주어야 한다.
인접행렬 adjMatrix
(dfs 함수)
- 총 4번 반복 (상하좌우)
- 다음 이동할 칸은 현재 좌표에 dy[i], dx[i]를 더한 값이다. (시계 방향으로 이동함)
- 이때 다음 이동할 칸이 아래 조건 중 하나라도 만족하면 continue로 건너뛴다.
1) 현재 좌표가 (0, 0), (n-1, m-1) 일 때 다음 이동하는 곳이 밖이면 안 된다.
2) 다음 이동할 칸이 0이라면 갈 수 없다.
3) 이미 방문한 곳은 또 가지 않는다. - 모두 해당하지 않으면, 방문 처리를 하고 다음 이동할 칸을 넣어 dfs 함수를 호출한다.
(입력 및 선언)
- 테스트 케이스 입력 받기
- 테스트 케이스 개수만큼 가로, 세로, 배추 위치를 입력 받는다.
- 배추 위치가 입력 되면 해당 위치의 adjMatrix에 1을 넣는다. (배추밭 만드는 과정)
- 방문 처리 배열도 2차원으로 선언하고 모두 False로 초기화시킨다.
- 지렁이 개수를 세는 cnt 변수도 여기에 넣어서 테스트케이스가 새로 시작할 때마다 0으로 초기화될 수 있도록 한다.
(배추 위치 찾기)
- 밭 전체를 돌기 위해 이중 반복문을 사용한다.
- 만약 배추밭(adjMatrix)가 1이고, 방문 처리가 안 된 곳이라면
- 방문 처리를 해주고, bfs 함수를 호출한다.
- 지렁이 개수(cnt)를 하나 늘린다.
(최종 출력)
- 지렁이 개수(cnt)를 출력한다.
최종 코드
BFS
from collections import deque
dy = [-1, 0, 1, 0]
dx = [0, 1, 0, -1]
# bfs - queue
def bfs() :
while bfsQueue :
currentLocation = bfsQueue.popleft()
curY = currentLocation[0]
curX = currentLocation[1]
for i in range(4) :
ny = curY + dy[i]
nx = curX + dx[i]
if ny < 0 or nx < 0 or ny >= n or nx >= m :
continue
if adjMatrix[ny][nx] == 0 :
continue
if visited[ny][nx] != 0 :
continue
visited[ny][nx] = True
bfsQueue.append((ny, nx))
t = int(input())
for _ in range(t) :
m, n, k = list(map(int, input().split()))
adjMatrix = [[0 for _ in range(m)] for _ in range(n)]
visited = [[False for _ in range(m)] for _ in range(n)]
bfsQueue = deque([])
cnt = 0
for _ in range(k) :
x, y = list(map(int, input().split()))
adjMatrix[y][x] = 1
# bfs - visited, queue, bfs 호출
for i in range(n) :
for j in range(m) :
if adjMatrix[i][j] == 1 and visited[i][j] == 0 :
bfsQueue.append((i, j))
visited[i][j] = True
bfs()
cnt += 1
print(cnt)DFS
import sys
sys.setrecursionlimit(10000) # 재귀 깊이 설정
dy = [-1, 0, 1, 0]
dx = [0, 1, 0, -1]
# dfs - Recursion
def dfs(y, x) :
global cnt
for i in range(4) :
ny = y + dy[i]
nx = x + dx[i]
if ny < 0 or nx < 0 or ny >= n or nx >= m :
continue
if adjMatrix[ny][nx] == 0 :
continue
if visited[ny][nx] != 0 :
continue
visited[ny][nx] = 1
dfs(ny, nx)
t = int(input())
for _ in range(t) :
m, n, k = list(map(int, input().split()))
adjMatrix = [[0 for _ in range(m)] for _ in range(n)]
visited = [[False for _ in range(m)] for _ in range(n)]
cnt = 0
for _ in range(k) :
x, y = list(map(int, input().split()))
adjMatrix[y][x] = 1
# dfs - visited, dfs
for i in range(n) :
for j in range(m) :
if adjMatrix[i][j] == 1 and visited[i][j] == 0 :
visited[i][j] = True
dfs(i, j)
cnt += 1
print(cnt)피드백
y축이 n
x축이 m
세로가 n, 가로가 m
둘이 너무 헷갈린다 ...
I mean