[프로그래머스] 정수삼각형 파이썬 풀이

[문제 링크]

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

[입출력 예]

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

def solution(triangle):
    answer = [] #1
    #answer의 i번째 원소에는 i번째 줄 까지 왔을 때 각 위치에서 누적값의 최대값

    for line_index, line in enumerate(triangle):
        answer.append([])
        for element_index,element in enumerate(line):
            if line_index ==0:
                answer[0].append(element) #2
                # 처음 원소는 그냥 추가

            if line_index >0 \
            and (element_index == 0 or element_index == len(line)-1): #3
                # 한 줄 안에서 맨 왼쪽이거나 맨 오른쪽일 경우
                # 값을 물려받을 경로가 한 군데 밖에 없음.
                  answer[line_index].append(element+answer[line_index-1][0] \
                                            if element_index == 0 \
                                            else element+answer[line_index-1][element_index-1])
            else:
                answer[line_index].append(element+max(answer[line_index-1][element_index],answer[line_index-1][element_index-1])) #4
                # 가운데 있는 element일 경우 왼쪽 부모와 오른쪽 부모 중 높은 값과 자신을 합함.

    return max(answer[-1])