BOJ/백준 - 1932 정수삼각형 C++

왼쪽 끝, 오른쪽 끝만 신경써주면 되는 문제이다.

왼쪽 끝을 예로 들면, dp[i][j]를 구하기 위해서는 저장된 dp[i-1][j]값과 arr[i][j]의 값이 필요하다.

오른쪽 끝, 그 중간도 저장된 값+dp[i][j]에 같이 더해 저장할 arr[i][j]의 값이 필요하다.

 

그대로 짜주면 이런 코드.

#include <iostream>
#include <algorithm>
 
using namespace std;
 
int arr[501][501];
int dp[501][501];
 
int main(void) {
    int N;
    cin >> N;
 
    for (int i = 1; i <= N; i++) {
        for (int j = 1; j<=i; j++) {
            cin >> arr[i][j];
        }
    }
 
    dp[1][1] = arr[1][1];
 
    for (int i = 2; i <= N; i++) {
        for (int j = 1; j <= i; j++) {
            if (j == 1) { //왼쪽 끝.
                dp[i][j] = dp[i - 1][j] + arr[i][j];
            }
            if (j == i) { //오른쪽 끝.
                dp[i][j] = dp[i - 1][j - 1] + arr[i][j];
            }
            else {
                dp[i][j] = max(dp[i - 1][j - 1], dp[i - 1][j]) + arr[i][j];
            }
        }
    }
    
    int answer = 0;
    for (int i = 1; i <= N; i++) {
        answer = max(dp[N][i], answer);
    }
    cout << answer;
 
    return 0;
}