[프로그래머스] Level2, 땅따먹기 (Python)

문제

https://programmers.co.kr/learn/courses/30/lessons/12913?language=python3 

코딩테스트 연습 - 땅따먹기

 

코드

def solution(land):
    answer = 0
    for i in range(1,len(land)):
        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 max(land[-1])

DP를 이용하는 문제이다.

맨 위에 행은 냅둬도 되어서 1부터 시작한다.

자신의 열은 제외하고 가장 큰 값을 구한다.

 

다른 방식 코드

def solution(land):
    for i in range(1,len(land)):
        for j in range(len(land[0])):
            land[i][j] += max(land[i-1][:j] + land[i-1][j+1:])
    return max(land[len(land)-1])

자신의 열을 제외하기 위해서 슬라이싱을 사용하여 자신의 열만 제외한다.