Algorithm/DP

[백준] 1932 '정수 삼각형' 파이썬(python)

나는SOU 2021. 12. 24. 00:00

https://www.acmicpc.net/problem/1932

 

1932번: 정수 삼각형

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

www.acmicpc.net

n = int(input())
dp = []
for i in range(n):
    dp.append(list(map(int, input().split())))
k = 2
for i in range(1, n):
    for j in range(k):
        if j==0:
            dp[i][j] = dp[i][j] + dp[i-1][j]
        elif i == j:
            dp[i][j] = dp[i][j] + dp[i-1][j-1]
        else :
            dp[i][j] = max(dp[i-1][j-1], dp[i-1][j]) + dp[i][j]
    k+=1

print(max(dp[n-1]))

Point

  • dp테이블을 활용한 DP 문제
i/j 0 1 2 3 4
0 7        
1 3 8      
2 8 1 0    
3 2 7 4 4  
4 4 5 2 6 5
  • 합산을 해나가는 것으로서 합산에는 세 가지 경우로 나눌 수 있다.
  • 본래는 삼각형인 것을 표로 만들어 넣은 것으로 잊질 말아야 한다.
  • 첫번째는 j = 0이면 오른쪽 대각선 위 값을 가져오는 것. 즉 dp[i][j] = dp[i-1][j] + dp[i][j]
  • 두번째는 j = i 인 경우로서 왼쪽 대각선의 값을 가져오는 것. 즉 dp[i][j] = dp[i-1[j-1] + dp[i][j]
  • 마지막은 나머지 경우로서 중앙에 있는 값들인데 이들은 왼쪽 대각선 위 혹은 오른쪽 대각선 위 중 값이 큰 것을 골라야 한다. 따라서 dp[i][j] = max(dp[i-1][j], dp[i-1[j-1]) + dp[i][j]
  • 이번 dp 문제는 표를 만들어서 그에 대한 케이스로 나누어 푸는 문제였다.