[백준알고리즘]1932번 정수삼각형 파이썬Python

h3h3문제

맨 위층부터 시작해서 아래에 있는 수 중 하나를 선택하여 아래층으로 내려올 때,

이제까지 선택된 수의 합이 최대가 되는 경로를 구하는 프로그램을 작성하라.

아래층에 있는 수는 현재 층에서 선택된 수의 대각선 왼쪽 또는 대각선 오른쪽에 있는 것 중에서만 선택할 수 있다.

삼각형의 크기는 1 이상 500 이하이다. 삼각형을 이루고 있는 각 수는 모두 정수이며, 범위는 0 이상 9999 이하이다.

 

h3h3입출력

 

h3h3입력

첫째 줄에 삼각형의 크기 n(1 ≤ n ≤ 500)이 주어지고, 둘째 줄부터 n+1번째 줄까지 정수 삼각형이 주어진다.

h3h3출력

첫째 줄에 합이 최대가 되는 경로에 있는 수의 합을 출력한다.

 

예시를 보면 알겠지만, 먼저 몇줄을 쓸건지 수를 입력 받는다.

입력받은 수만큼 삼각형을 만든다.

그럼 출력으로 최댓값이 나옴을 확인할 수 있다.

num = int(input())
shape = [list(map(int, input().split())) for _ in range(num)]

for i in range(num - 1):
    for j in range(len(shape[i + 1])):
        if(j == 0):
            shape[i + 1][j] = shape[i][j] + shape[i + 1][j]
        elif(j == len(shape[i + 1]) - 1):
            shape[i + 1][-1] = shape[i][-1] + shape[i + 1][-1]
        else:
            shape[i + 1][j] = max(shape[i][j - 1] + shape[i + 1][j], shape[i][j] + shape[i + 1][j])
print(max(shape[-1]))