프로그래머스_정수 삼각형_python

2021. 8. 29. 16:41코딩테스트

반응형

 

https://programmers.co.kr/learn/courses/30/lessons/43105

 

코딩테스트 연습 - 정수 삼각형

[[7], [3, 8], [8, 1, 0], [2, 7, 4, 4], [4, 5, 2, 6, 5]] 30

programmers.co.kr

def solution(triangle):
    i_len = len(triangle)
    a = [ len(t)*[0] for t in triangle]
    def f(i, j):
        if i == i_len:
            return 0
        j_len = len(triangle[i])
        if j == j_len:
            return 0
        elif a[i][j] != 0:
            return a[i][j]
        a[i][j] = triangle[i][j]+max(f(i+1,j), f(i+1,j+1))
        return a[i][j]
    answer = f(0,0)
    return answer

[내코드]

탑다운. 

삼각형과 동일한 크기 리스트 생성, 0으로 초기화

맨위를 호출하면 왼쪽, 오른쪽중 큰 숫자를 택하여 삼각형값과 합침

 

def solution(triangle):
    dp = []
    for t in range(1, len(triangle)):
        for i in range(t+1):
            if i == 0:
                triangle[t][0] += triangle[t-1][0]
            elif i == t:
                triangle[t][-1] += triangle[t-1][-1]
            else:
                triangle[t][i] += max(triangle[t-1][i-1], triangle[t-1][i])
    return max(triangle[-1])

[다른사람코드]

아래방향으로 for문을 돌리며 위의 값을 현재값에 더함.

마지막 줄의 max값을 리턴

반응형