알고리즘 공부/COS Pro 1급 모의고사 답안

파트2. 한 줄 바꾸기 - 종이접기

철매존 2021. 11. 27. 23:15
728x90

문제 설명

4 x 4 크기인 정사각형 종이가 1 x 1 크기인 격자 칸으로 나누어져 있습니다. 이 종이를 가로축 혹은 세로축에 평행한 격자 선을 따라 한 번 접었을 때, 만나는 격자 칸에 적힌 숫자의 합이 최대가 되도록 하려 합니다. 종이를 접을 때는 만나는 격자 칸이 정확히 일치하도록 해야 합니다.

예를 들어 다음과 같이 4 x 4 크기인 종이가 있을 때,

종이는 점선 중 하나를 따라서 접을 수 있습니다. 이때, 붉은색 점선을 따라 종이를 접으면 36과 19가 적힌 칸이 정확히 만납니다. 두 숫자의 합은 55이며, 이때가 최댓값입니다.

4 x 4 크기인 정사각형 종이의 각 격자 칸에 적힌 숫자가 담긴 배열 grid가 매개변수로 주어질 때, 종이를 접었을 때 만나게 되는 격자 칸에 적힌 숫자의 합 중 최댓값을 return 하도록 solution 메소드를 작성했습니다. 그러나, 코드 일부분이 잘못되어있기 때문에, 몇몇 입력에 대해서는 올바르게 동작하지 않습니다. 주어진 코드에서 한 줄만 변경해서 모든 입력에 대해 올바르게 동작하도록 수정하세요.


매개변수 설명

4 x 4 크기인 정사각형 종이의 각 격자 칸에 적힌 숫자가 담긴 배열 grid가 solution 메소드의 매개변수로 주어집니다.

  • 각 격자칸에 적힌 수는 1 이상 100 이하인 자연수입니다.

return 값 설명

격자 선을 따라 종이를 한 번 접었을 때 만나는 격자 칸에 적힌 숫자의 합 중 최댓값을 return 해주세요.

  • 격자 선은 문제의 예제와같이 가로, 혹은 세로 방향으로 평행한 점선을 말합니다.

예제gridreturn
[[1, 4, 16, 1], [20, 5, 15, 8], [6, 13, 36, 14], [20, 7, 19, 15]] 55
예제 설명

문제의 예제와 같습니다.

 


 

class Solution {
    public int solution(int[][] grid) {
        int answer = 0;
        for(int i = 0; i < 4; i++)
            for(int j = 0; j < 4; j++)
                for(int k = j + 1; k < 4; k += 2)
                    answer = Math.max(answer, Math.max(grid[i][j] + grid[i][k], grid[j][i] + grid[k][i]));
        return answer;
    }

    // 아래는 테스트케이스 출력을 해보기 위한 main 메소드입니다. 아래에는 잘못된 부분이 없으니 위의 코드만 수정하세요.
    public static void main(String[] args) {
        Solution sol = new Solution();
        int[][] grid = {{1, 4, 16, 1}, {20, 5, 15, 8}, {6, 13, 36, 14}, {20, 7, 19, 15}};
        int ret = sol.solution(grid);

        // [실행] 버튼을 누르면 출력 값을 볼 수 있습니다.
        System.out.println("solution 메소드의 반환 값은 " + ret + " 입니다.");
    }
}

k를 인자로 받는 for문이 실제 종이를 접는 모든 경우이다.

k+=2 인 이유는 종이를 접는경우를 실제로 생각해 보면 알 수 있다. 

(0, 0)위치를 기준으로 가로로 접는 경우 (1,0) / (3,0)만 가능하니까 그렇다.

 

(0,0)을 기준으로 종이를 접는 모든 경우는

가로

   (0,0) + (0,1) / (0,0) + (0,3)

세로

   (0,0) + (1,0) / (0,0) + (3,0)

이다.

 

근데 어째 이 문제는 k가 j를 기준으로는 나오는데, i를 기준으로 하는건 안나온다. 그래서 다른 방식으로 구현한다.

앞서 말했듯이 k는 j를 기준으로 움직인다. 그래서 j가 배열의 x축에 넣는 경우, y축에 넣는 경우 두 가지를 넣고 k도 그걸 따라가면 가로 세로를 한번에 구할 수 있다.

예를 들어

arr[i][j] + arr[i][k] -> 가로쪽을 구하는법

arr[j][i] + arr[k][i] -> 세로쪽을 구하는법

이렇게 해서 for문을 전체를 돌리면 된다.