철갑이의 이모저모

[Programmers] 정수 삼각형 with Python 본문

알고리즘

[Programmers] 정수 삼각형 with Python

철갑 2020. 12. 2. 11:58
728x90

문제

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):
    for i in range(1,len(triangle)):
        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])
    return max(triangle[-1])

728x90