[프로그래머스] 정수 삼각형 - Python

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

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

     

        7                               7                            7                            7                              7

      3   8                         10  15                     10  15                     10  15                        10  15

    8   1   0          ->       8   1   0         ->      18  16  15     ->        18  16  15        ->        18  16  15  

  2   7   4   4              2   7   4   4               2    7    4    4           20  25  20  19              20  25  20  19

4   5   2   6   5        4   5   2   6   5          4    5    2    6    5       4    5    2    6    5         25  30  27  26  24

 

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