PS/Programmers

Programmers / Level 2 / 땅따먹기 / C++

KimMinJun 2022. 2. 10. 13:41

문제 설명

땅따먹기 게임을 하려고 합니다. 땅따먹기 게임의 땅(land)은 총 N행 4열로 이루어져 있고,
모든 칸에는 점수가 쓰여 있습니다.
1행부터 땅을 밟으며 한 행씩 내려올 때, 각 행의 4칸 중 한 칸만 밟으면서 내려와야 합니다. 
단, 땅따먹기 게임에는 한 행씩 내려올 때, 같은 열을 연속해서 밟을 수 없는 특수 규칙이 있습니다.

예를 들면,

| 1 | 2 | 3 | 5 |

| 5 | 6 | 7 | 8 |

| 4 | 3 | 2 | 1 |

로 땅이 주어졌다면, 1행에서 네번째 칸 (5)를 밟았으면, 2행의 네번째 칸 (8)은 밟을 수 없습니다.

마지막 행까지 모두 내려왔을 때, 얻을 수 있는 점수의 최대값을 return하는 solution 함수를 완성해 주세요.
위 예의 경우, 1행의 네번째 칸 (5), 2행의 세번째 칸 (7), 3행의 첫번째 칸 (4) 땅을 밟아 16점이 최고점이 되므로
16을 return 하면 됩니다.

 

제한사항
  • 행의 개수 N : 100,000 이하의 자연수
  • 열의 개수는 4개이고, 땅(land)은 2차원 배열로 주어집니다.
  • 점수 : 100 이하의 자연수
입출력 예
land answer
[[1,2,3,5],[5,6,7,8],[4,3,2,1]] 16

 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

int solution(vector<vector<int> > land) {
    int answer = 0;
    vector<vector<int> > dp(land.size(), vector<int>(4));
    
    for(int i=0; i<4; i++) {
        dp[0][i] = land[0][i];
    }    
    
    for(int i=1; i<dp.size(); i++) {
        for(int j=0; j<4; j++) {
            int max = 0;
            for(int k=0; k<4; k++) {
                if(j == k) continue;
                if(max < dp[i-1][k]) {
                    max = dp[i-1][k];
                }
            }
            dp[i][j] = land[i][j] + max;
        }
    }
    
    answer = *max_element(dp.back().begin(), dp.back().end());

    return answer;
}

이 문제는 dp로 해결해주어야 한다.

dp로 해당하는 인덱스 까지 갈때의 최대 누적합을 구해주면 결국 마지막행에서 최댓값을 구해주면 그것이 답이된다.

 

dp의 크기와, dp의 맨 첫번째 행은 land와 똑같이 지정해준다.

첫번째 행은 인덱스가 0이므로 누적할 값이 없기 때문이다.

 

예를들어 dp[1][3]을 구한다고 하면 아래의 3개중에서 최댓값으로 저장해주면 된다.

  1. dp[0][1] + land[1[[3]
  2. dp[0][2] + land[1[[3]
  3. dp[0][4] + land[1[[3]

 

문제에서 같은 열은 지날 수 없기 때문에 제외해준다.

 

이런 방식으로 마지막 행까지 누적합을 구해준다.

그러면 마지막 행에서 최댓값을 구해주면, 결국 마지막 행까지 갈때의 경로중에서 최댓값을 구하는것이므로 그 값을

return 해주면 된다.