[알고리즘] Swift -백준 # 1743 (음식물 피하기) -DFS

문제링크

https://www.acmicpc.net/problem/1743

1743번: 음식물 피하기

 

문제

코레스코 콘도미니엄 8층은 학생들이 3끼의 식사를 해결하는 공간이다. 그러나 몇몇 비양심적인 학생들의 만행으로 음식물이 통로 중간 중간에 떨어져 있다. 이러한 음식물들은 근처에 있는 것끼리 뭉치게 돼서 큰 음식물 쓰레기가 된다. 

이 문제를 출제한 선생님은 개인적으로 이러한 음식물을 실내화에 묻히는 것을 정말 진정으로 싫어한다. 참고로 우리가 구해야 할 답은 이 문제를 낸 조교를 맞추는 것이 아니다. 

통로에 떨어진 음식물을 피해가기란 쉬운 일이 아니다. 따라서 선생님은 떨어진 음식물 중에 제일 큰 음식물만은 피해 가려고 한다. 

선생님을 도와 제일 큰 음식물의 크기를 구해서 “10ra"를 외치지 않게 도와주자.

입력

첫째 줄에 통로의 세로 길이 N(1 ≤ N ≤ 100)과 가로 길이 M(1 ≤ M ≤ 100) 그리고 음식물 쓰레기의 개수 K(1 ≤ K ≤ N×M)이 주어진다.  그리고 다음 K개의 줄에 음식물이 떨어진 좌표 (r, c)가 주어진다.

좌표 (r, c)의 r은 위에서부터, c는 왼쪽에서부터가 기준이다. 입력으로 주어지는 좌표는 중복되지 않는다.

출력

첫째 줄에 음식물 중 가장 큰 음식물의 크기를 출력하라.

 

---

분석

음식물이 떨어져 근처에 있는 것끼리 뭉쳐 큰 쓰레기가 된다.

가장 큰 음식물을 피하자

 

풀이 과정

가로세로길이가 주어진다. n*m의 사이즈의 행렬

주어진 좌표에 값을 넣고

탐색을 통해 쓰레기가 있는 곳에 동서남북으로 1칸씩(인접한 칸)에 쓰레기가 존재한다면 해당 위치에서 다시 탐색 동서남북으로 탐색 (재귀)

재귀를 부를 때 count +1

결과는 count의 최댓값

 

 

풀이

let num = readLine()!.split(separator: " ").map{ Int($0)!}
let n = num[0]
let m = num[1]
let k = num[2]

var checkArr = [[Bool]](repeating: Array(repeating: false, count: m+1), count: n+1)
var arr = [[Bool]](repeating: Array(repeating: false, count: m+1), count: n+1)

for _ in 0..<k {
    let input = readLine()!.split(separator: " ").map{ Int($0)!}
    let r = input[0]
    let c = input[1]
    
    arr[r][c] = true
}

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

func isValid(y: Int, x: Int) -> Bool {
    return 1 <= y && y <= n && 1 <= x && x <= m ? true : false
}

func dfs(r: Int, c: Int) -> Int {
    checkArr[r][c] = true
    var count = 1
    for k in 0..<4 {
        let nextY = r + dy[k]
        let nextX = c + dx[k]
        
        if isValid(y: nextY, x: nextX) && !checkArr[nextY][nextX] && arr[nextY][nextX] {
            count += dfs(r: nextY, c: nextX)
        }
    }
    return count
}

var cnt = 0
for i in 1...n {
    for j in 1...m {
        if arr[i][j] {
            cnt = max(dfs(r: i, c: j), cnt)
        }
    }
}
print(cnt)

 

결과

 

 

알게된 점

count를 반환하는 메소드를 만들어서 최댓값을 찾는 방법