[Java] 백준 2169 로봇 조종하기

hyunnzl·2025년 1월 23일

백준

목록 보기
40/116
post-thumbnail

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

난이도

골드2

문제

NASA에서는 화성 탐사를 위해 화성에 무선 조종 로봇을 보냈다. 실제 화성의 모습은 굉장히 복잡하지만, 로봇의 메모리가 얼마 안 되기 때문에 지형을 N×M 배열로 단순화 하여 생각하기로 한다.

지형의 고저차의 특성상, 로봇은 움직일 때 배열에서 왼쪽, 오른쪽, 아래쪽으로 이동할 수 있지만, 위쪽으로는 이동할 수 없다. 또한 한 번 탐사한 지역(배열에서 하나의 칸)은 탐사하지 않기로 한다.

각각의 지역은 탐사 가치가 있는데, 로봇을 배열의 왼쪽 위 (1, 1)에서 출발시켜 오른쪽 아래 (N, M)으로 보내려고 한다. 이때, 위의 조건을 만족하면서, 탐사한 지역들의 가치의 합이 최대가 되도록 하는 프로그램을 작성하시오.

입력

첫째 줄에 N, M(1≤N, M≤1,000)이 주어진다. 다음 N개의 줄에는 M개의 수로 배열이 주어진다. 배열의 각 수는 절댓값이 100을 넘지 않는 정수이다. 이 값은 그 지역의 가치를 나타낸다.

출력

첫째 줄에 최대 가치의 합을 출력한다.

내 코드

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

class Main {
	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[][] 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());
			}
		}

		dp[0][0] = map[0][0];
		for (int j = 1; j < M; j++) {
			dp[0][j] = map[0][j] + dp[0][j - 1];
		}

		for (int i = 1; i < N; i++) {

			int[] left = new int[M];
			left[0] = dp[i - 1][0] + map[i][0];
			for (int j = 1; j < M; j++) {
				left[j] = Math.max(left[j - 1], dp[i - 1][j]) + map[i][j];
			}

			int[] right = new int[M];
			right[M - 1] = dp[i - 1][M - 1] + map[i][M - 1];
			for (int j = M - 2; j >= 0; j--) {
				right[j] = Math.max(right[j + 1], dp[i - 1][j]) + map[i][j];
			}

			for (int j = 0; j < M; j++) {
				dp[i][j] = Math.max(left[j], right[j]);
			}
		}
		System.out.println(dp[N - 1][M - 1]);
	}
}

왼쪽으로도 갈 수 있고, 오른쪽으로도 갈 수 있기 때문에 dp를 각각 확인해서 최적의 길을 찾아야 한다.

제일 윗 줄은 왼쪽값과 비교해서 오른쪽으로만 나아갈 수 있기 때문에 양방향을 확인해주지 않아도 된다.

두번째 줄 부터는 왼쪽에서 출발하는 것과 오른쪽에서 출발하는 것 두 방향을 확인해줘야 한다.

각 방향으로 확인해주고 마지막으로 dp에 업데이트 해주면 마지막 위치가 최적의 값이 된다.


0개의 댓글