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

5348 단어 pythonprogrammersDPDP

문제

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

위와 같은 삼각형의 꼭대기에서 바닥까지 이어지는 경로 중, 거쳐간 숫자의 합이 가장 큰 경우를 찾아보려고 합니다. 아래 칸으로 이동할 때는 대각선 방향으로 한 칸 오른쪽 또는 왼쪽으로만 이동 가능합니다. 예를 들어 3에서는 그 아래칸의 8 또는 1로만 이동이 가능합니다.

삼각형의 정보가 담긴 배열 triangle이 매개변수로 주어질 때, 거쳐간 숫자의 최댓값을 return 하도록 solution 함수를 완성하세요.

제한 사항

  • 삼각형의 높이는 1 이상 500 이하입니다.
  • 삼각형을 이루고 있는 숫자는 0 이상 9,999 이하의 정수입니다.

입출력 예시

아이디어

✅ 삼각형을 차례로 돌면서 그 때의 합을 구해 triangle을 업데이트 한다.

  1. 회전을 하다가 제일 왼쪽에 있는 경우는 오른쪽 위에서 오는 경우만 있으므로 원래 숫자에 더해준다.
  2. 제일 오른쪽에 있는 경우는 왼쪽 위에서 오는 경우만 있으므로 원래 숫자에 더해준다.
  3. 끝이 아닌 가운데에 있는 경우, 양쪽 위에서 모두 오기 때문에 왼쪽과 오른쪽 중 큰 수를 더해준다.
  4. 마지막에 가장 큰 수를 출력하면 된다.

solution함수_python

def solution(triangle):
    for i in range(1, len(triangle)):
        for j in range(len(triangle[i])):
            if j == 0:  #left
                triangle[i][j] += triangle[i-1][j]
            elif j == len(triangle[i])-1:   #right
                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])

좋은 웹페이지 즐겨찾기