[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 이하의 자연수

 

 

class Solution {
    int solution(int[][] land) {
        int len = land.length-1;

        for(int l = 1; l < land.length; l++) {
            land[l][0] += Math.max(land[l-1][1], Math.max(land[l-1][2], land[l-1][3]));
            land[l][1] += Math.max(land[l-1][0], Math.max(land[l-1][2], land[l-1][3]));
            land[l][2] += Math.max(land[l-1][0], Math.max(land[l-1][1], land[l-1][3]));
            land[l][3] += Math.max(land[l-1][0], Math.max(land[l-1][1], land[l-1][2]));
        }

        return Math.max(land[len][0], Math.max(land[len][1], Math.max(land[len][2], land[len][3])));
    }
}

DP를 통한 문제 해결법이다.

두번째 줄 부터 시작을 해서, 이전 줄의 공통된 칸이 아닌 칸들을 더해가면서 그 중 가장 큰 값을 해당 줄과 칸에 저장을 하는 방식이다.

 

코딩테스트이다보니 문제에 4열로 이루어졌으므로 0, 1, 2, 3 으로 해도 되지만, 이건 이 문제에 특정되는 코드라는 점이 뭔가 거슬렸다.

 

그래서 열이 4개가 아니라 여러 열이라는 가정하에 다시 코드를 구현했다.

class Solution {
    int solution(int[][] land) {
        int answer = 0;
        int landLen = land[0].length;

        int x = 1, landIdx = 0;
        while(x < land.length) {
            int maxScore = 0;
            for(int l = 0; l < landLen; l++) {
                if(l != landIdx) {
                    maxScore = Math.max(maxScore, land[x-1][l]);
                }
            }
            land[x][landIdx++] += maxScore;
            
            if(landIdx == landLen) {
                landIdx = 0;
                x++;
            }
        }
        
        for(int i = 0; i < landLen; i++) {
            answer = Math.max(answer, land[land.length-1][i]);
        }

        return answer;
    }
}

당연히 열이 여러개라는 가정하에 구현한 코드이니 열이 4개라는 특정을 짓고 해결한 코드보단 시간이 더 오래걸린다.

(생각보다 막 그렇게 오래걸리는거 같진 않다)

 

아래 코드는 괜히 DFS 연습을 하고싶어서 DFS로 해결하려고 구현한 코드이다.

class Solution {
    
    int solution(int[][] land) {
        return dfs(land, 0, -1, 0);
    }
    
    public int dfs(int[][] land, int x, int prevLand, int score) {
        if(x == land.length) {
            return score;
        }
        
        int maxScore = 0;
        for(int y = 0; y < land[0].length; y++) {
            if(y != prevLand) {
                int currentScore = score + land[x][y];
                int resultScore = dfs(land, x+1, y, currentScore);
                maxScore = Math.max(maxScore, resultScore);
            }
        }
        
        return maxScore;
    }
}

 

다만, 모든 테스트에 시간초과가 발생했다... 테스트가 통과되는것을 봐선 왠지 시간만 생각 안한다면 제대로 동작하는 코드라고 생각한다.

(역시 모든 경우의수를 확인하는 작업은 시간이 너무 오래걸린다)

 

 

출처

https://school.programmers.co.kr/learn/courses/30/lessons/12913