본문 바로가기

algorithm/Programmers

프로그래머스 Level 3. 정수 삼각형

문제 설명

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

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

제한사항

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

입출력 예

triangle result

[[7], [3, 8], [8, 1, 0], [2, 7, 4, 4], [4, 5, 2, 6, 5]] 30

출처

[

Sweden

International Olympiad in Informatics – Statistics Contact information Participation in IOI (based on database records) First participation: 1990Years participated: 30Contestants participated: 80 Perfomance in IOI Gold medals: 13Silver medals: 29Bronze m

stats.ioinformatics.org

](http://stats.ioinformatics.org/countries/SWE)

 

   0. 다이나믹 프로그래밍을 이용해서 접근을 해보자.

  1. d[i][j] = i행에서 j열에서 끝나는 최대값
  2. d[i][j] = 왼쪽 위 대각선과 오른쪽 위 대각선의 max 값을 triangle[i][j]를 더해준다.
  3. 왼쪽 위 대각선과 오른쪽 위 대각선이 있을 때와 없을 때를 고려해줘야 한다.
#include <string>
#include <vector>
#include <algorithm>
using namespace std;

int d[501][501];
int solution(vector<vector<int>> triangle) {
    int answer;
    int tri_size = triangle.size();

    for(int i = 1; i <= tri_size; i++){
        for(int j = 1; j <= i; j++){

            d[i][j] = d[i-1][j] + triangle[i - 1][j - 1]; // 오른쪽 위 대각선을 기본으로 더해준다.

            if(j > 1 &&d[i][j] < d[i - 1][j - 1] + triangle[i - 1][j - 1]){ // 왼쪽위 대각선이 있을 경우 max 값 넣기
                d[i][j] = d[i - 1][j - 1] + triangle[i - 1][j - 1];
            }     
        }  
    }

    answer = d[tri_size][0];
    for(int i = 1; i <= tri_size; i++){ // i행 j열에서 최댓값을 구한다.
        if(answer < d[tri_size][i]){
            answer = d[tri_size][i];
        }
    }
    return answer;
}