[실전 문제] 자원 채취

1. 문제

N x M의 지도가 주어지며, 이 지도의 각 칸에는 자원이 존재한다. 자원의 양은 정수로 나타난다. 다음 그림은 5 x 6 의 지도에 존재하는 자원을 나타낸다.

철수는 자원을 채취하는 로봇을 갖고 있으며, 이 로봇은 (0, 0) 에서 출발하여 (N-1, M-1) 에서 자원 채취를 마친다. 로봇은 한가지 제약이 있는데, 오른쪽과 아랫쪽으로밖에 움직일 수 없다는 것이다. 이 로봇을 이용하여 가장 많이 채취할 수 있는 자원의 양을 출력하는 프로그램을 작성하시오. 위의 예제의 경우 다음과 같이 채취하는 것이 최대이며, 그 양은 49이다.

---

입력

첫 번째 줄에 N, M이 주어진다. ( 1 ≤ N, M ≤ 1,000 ) 두 번째 줄부터 N x M 의 지도에 존재하는 자원의 양이 주어진다.

출력

로봇을 이용하여 채취할 수 있는 자원의 양의 최댓값을 출력한다.

예제 입력

5 6
1 7 3 2 8 0
9 2 3 4 5 4
3 4 7 8 2 2
1 4 3 1 4 1
3 2 5 5 3 8

예제 출력

49

 

 

 

2. 풀이

i가 행, j가 열이라고 할 때,

• i = 0 일 때, 위에서 올 수 있는 경로가 없기 때문에 오른쪽으로 가는 방향만 확인
• j = 0 일 때, 오른쪽에서 올 수 있는 경로가 없기 때문에 아래쪽으로만 가는 방향만 확인

 

import java.io.*;
import java.util.StringTokenizer;

public class Main {

    private static final BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    private static final BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

    public static void main(String[] args) throws IOException {

        StringTokenizer st = new StringTokenizer(br.readLine());
        int n = Integer.parseInt(st.nextToken());
        int m = Integer.parseInt(st.nextToken());

        int[][] arr = new int[n][m];

        for(int i = 0; i < n; i++) {
            int sum = 0;
            st = new StringTokenizer(br.readLine());
            for(int j = 0; j < m; j++) {
                int value = Integer.parseInt(st.nextToken());
                if(i == 0) {
                    sum += value;
                    arr[i][j] = sum;
                } else if(j == 0) {
                    arr[i][j] = arr[i-1][j] + value;
                } else {
                    arr[i][j] = Math.max(arr[i-1][j], arr[i][j-1]) + value;
                }
            }
        }
        bw.write(arr[n-1][m-1] + "");

        br.close();
        bw.flush();
        bw.close();
    }
}