BOJ-11048 이동하기 Swift

문제

준규는 N×M 크기의 미로에 갇혀있다. 미로는 1×1크기의 방으로 나누어져 있고, 각 방에는 사탕이 놓여져 있다. 미로의 가장 왼쪽 윗 방은 (1, 1)이고, 가장 오른쪽 아랫 방은 (N, M)이다.

준규는 현재 (1, 1)에 있고, (N, M)으로 이동하려고 한다. 준규가 (r, c)에 있으면, (r+1, c), (r, c+1), (r+1, c+1)로 이동할 수 있고, 각 방을 방문할 때마다 방에 놓여져있는 사탕을 모두 가져갈 수 있다. 또, 미로 밖으로 나갈 수는 없다.

준규가 (N, M)으로 이동할 때, 가져올 수 있는 사탕 개수의 최댓값을 구하시오.

입력

첫째 줄에 미로의 크기 N, M이 주어진다. (1 ≤ N, M ≤ 1,000)

둘째 줄부터 N개 줄에는 총 M개의 숫자가 주어지며, r번째 줄의 c번째 수는 (r, c)에 놓여져 있는 사탕의 개수이다. 사탕의 개수는 0보다 크거나 같고, 100보다 작거나 같다.

출력

첫째 줄에 준규가 (N, M)으로 이동할 때, 가져올 수 있는 사탕 개수를 출력한다.

내가 푼 풀이

- N x M 크기의 배열이 주어지면 (1,1) 에서 (N, M)으로 이동할 때 오른쪽, 아래, 오른쪽대각선 방향으로만 이동한다.

- dp배열을 (n+1) x (m+1) 크기만큼 선언해서 dp[i][j] = i, j까지의 사탕 개수 최댓값으로 저장한다.

- 이는 dp[i][j] = max( dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + arr[i][j]로 나타낼 수 있다.

 

코드로 구현하면 다음과 같다.

 

import Foundation

let input = readLine()!.split(separator: " ").map{ Int($0)! }
var dp = Array(repeating: Array(repeating: 0, count: input[1]+1), count: input[0]+1)
var arr = Array(repeating: Array(repeating: 0, count: input[1]+1), count: input[0]+1)

// 사탕 미로 입력
for i in 0..<input[0] {
    var a = readLine()!.split(separator: " ").map{ Int($0)! }
    for j in 0..<a.count {
        arr[i+1][j+1] = a[j]
    }
}

// dp 입력
dp[1][1] = arr[1][1]
for i in 1...input[0] {
    for j in 1...input[1] {
        dp[i][j] = max(dp[i-1][j], dp[i][j-1], dp[i-1][j-1]) + arr[i][j]
    }
}
print(dp[input[0]][input[1]])