프로그래머스 - 땅따먹기(Swift)

프로그래머스 - 땅따먹기(Swift)

 

 

 문제 설명

 

https://school.programmers.co.kr/learn/courses/30/lessons/12913

프로그래머스

 

 

 다른 사람의 풀이

 

DP유형에 익숙하지 않아서 그런지 DP생각이 떠오르지 않아서 다른사람의 풀이를 참고했다.

 

처음에는 n의 범위가 100,000이기 때문에 웬만하면 한 번만 for문을 거쳐야겠다고 생각했다. (이제는 DP를 떠올려보자)

 

1 2 3 4

5 6 7 8 로 되어있을때

2번째 행의 5를 밟으려면 첫번째 행의 1을 제외한 2, 3, 4 중의 최대값을 밟아야하고, 

2번째 행의 6을 밟으려면 첫번째 행의 2를 제외한 1, 3, 4 중의 최대값을 밟아야한다.

 

즉 다음행에서 전의 열을 제외하고 열마다 값을 누적시키면 최대를 구할 수 있다.

여기서 새로 알게 된 것은 스위프트 내장 함수인 max에는 여러개의 인자가 들어올 수 있다는 것이다.

 

import Foundation

func solution(_ land:[[Int]]) -> Int{
    var answer = 0
    var land = land
    for i in 1..<land.count {
        land[i][0] += max(land[i-1][1], land[i-1][2], land[i-1][3])
        land[i][1] += max(land[i-1][0], land[i-1][2], land[i-1][3])
        land[i][2] += max(land[i-1][0], land[i-1][1], land[i-1][3])
        land[i][3] += max(land[i-1][0], land[i-1][1], land[i-1][2])
    }
    return land.last!.max()!
}