[백준 15666번] 로봇 조종하기

https://www.acmicpc.net/problem/2169

 

package dp;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class DP2169 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        int N = Integer.parseInt(st.nextToken());
        int M = Integer.parseInt(st.nextToken());

        int[][] map = new int[N][M];
        int[][] temp = new int[2][M];
        int[][] dp = new int[N][M];

        for (int i = 0; i < N; i++) {
            st = new StringTokenizer(br.readLine());
            for (int j = 0; j < M; j++) {
                map[i][j] = Integer.parseInt(st.nextToken());
            }
        }

        for (int i = 0; i < N; i++) {
            for (int j = 0; j < M; j++) {
                if (i == 0) {
                    if (j == 0) {
                        dp[i][j] = map[i][j];
                    } else {
                        dp[i][j] = dp[i][j - 1] + map[i][j];
                    }
                } else {
                    if (j == 0) {
                        temp[0][j] = dp[i - 1][j] + map[i][j];
                        temp[1][M - 1 - j] = dp[i - 1][M - 1] + map[i][M - 1];
                    } else {
                        temp[0][j] = Math.max(dp[i - 1][j], temp[0][j - 1]) + map[i][j];
                        temp[1][M - 1 - j] = Math.max(dp[i - 1][M - 1 - j], temp[1][M - j]) + map[i][M - 1 - j];
                    }
                }
            }
            if (i > 0) {
                for (int k = 0; k < M; k++) {
                    dp[i][k] = Math.max(temp[0][k], temp[1][k]);
                }
            }

        }


        System.out.println(dp[N - 1][M - 1]);
    }
}

 

DP 역사상 최고로 어려운 문제였다! OO 코테에서 유사한 유형이 출제돼서, 복습 차원에서 직접 풀어봤다.

1. 첫 줄 초기화
   첫 번째 줄의 값을 모두 더해서 DP 배열의 초기값으로 설정한다. 이때 방향 기준은 왼쪽에서 오른쪽으로 진행된다.
2. 아래로 이동하는 경우 계산 (왼쪽 아래)
   그 다음 줄로 내려가면서 가능한 이동을 고려한다. 먼저 왼쪽 아래 방향으로 이동하는 경우를 temp[0] 배열에 저장한다. 예를 들어, (1, 0)은 (0, 0)에서 내려온 값에 자기 자신의 값을 더해 저장한다.
   이후 (1, 1)은 이전 값 (1, 0)에서 온 값과 바로 위의 (0, 1)에서 내려온 값을 비교하여 더 큰 값을 선택해 temp에 저장한다. 이 과정을 계속 반복한다.
3. 오른쪽 아래 방향 처리
   다음으로 오른쪽 아래 방향으로 이동하는 경우를 처리한다. 방식은 2번과 동일하며 이번에 temp[1] 배열에 저장한다. 즉, 이전 단계에서 온 값과 바로 위에서 내려온 값을 비교해 더 큰 값을 선택한다.
4. 최댓값 갱신
   같은 위치에 대해 왼쪽 아래와 오른쪽 아래에서 온 값 중 더 큰 값을 dp 배열에 저장한다. 이렇게 하면 한 줄이 완성된다.
5. 반복
   이 과정을 모든 줄에 대해 반복하면 최종적으로 최대 값을 구할 수 있다.