💡코딩테스트/프로그래머스

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

두_두 2023. 5. 16. 14:38

동적계획법(Dynamic Programming) | DP

🌏 문제

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

👽 풀이 1

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

    
    return max(triangle[-1])

 

👽 풀이 2

def solution(triangle):
    dp = [[0] * len(triangle) for _ in range(len(triangle))]
    dp[0][0] = triangle[0][0]
    
    for i in range(0, len(triangle) - 1): 
        for j in range(len(triangle[i])): 
            dp[i + 1][j] = max(dp[i + 1][j], dp[i][j] + triangle[i + 1][j]) 
            dp[i + 1][j + 1] = max(dp[i + 1][j + 1], dp[i][j] + triangle[i + 1][j + 1]) 
    
    return max(dp[-1])

 

728x90