https://school.programmers.co.kr/learn/courses/30/lessons/12913?language=java
땅따먹기 게임을 하려고 합니다. 땅따먹기 게임의 땅(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 |
입출력 예 설명
입출력 예 #1
문제의 예시와 같습니다.
import java.util.*;
class Solution {
int solution(int[][] land) {
int[][] dp = new int[land.length+1][4];
for(int i = 1; i<dp.length; i++){
for(int j = 0; j<4; j++){
for(int d=0; d<4; d++){
if(d==j) continue;
dp[i][j] = Math.max(dp[i-1][d]+land[i-1][j], dp[i][j]);
}
}
}
int ans=0;
for(int j=0; j<4; j++){
if(dp[dp.length-1][j]>ans)
ans = dp[dp.length-1][j];
}
return ans;
}
}
처음에는 조합이라고 생각해서 연속되지 않는 열 조합을 뽑아내어 풀었는데 전부 시간초과가 나버렸다.
결국 구글링 해버렸지만, DP를 이용한다는 아이디어만 얻었을 뿐 코드까지 베끼진 않았다.
//연속 없는 중복 조합으로 열 조합 생성
//모든 열 조합의 합을 구하여 최대값 계산
class Solution {
int len;
int[] result = null;
int[][] coppied;
int ans = 0;
int solution(int[][] land) {
coppied = land;
// System.out.println(Arrays.toString(coppied));
len = land.length;
result = new int[len];
combi(0,0);
return ans;
}
void combi(int idx, int before){
// 조합 끝
if(idx==len){
int sum = 0;
for(int i=0; i<len; i++){
sum += coppied[i][result[i]];
}
if(sum > ans)
ans = sum;
return;
}
// 조합 시작
for(int i=0; i<4; i++){
if(i==before)
continue;
result[idx] = i;
combi(idx+1, i);
}
}
}
조합으로 풀어버린 멍청한 코드 공개.
생각해보니 몇달 전에 비슷한 문제를 풀었던 적이 있다....
난 또 까먹어 버린 것이다..