백준 알고리즘

[Silver I] 점프 - 1890

성능 요약

메모리: 13156 KB, 시간: 8 ms

분류

다이나믹 프로그래밍

문제 설명

N×N 게임판에 수가 적혀져 있다. 이 게임의 목표는 가장 왼쪽 위 칸에서 가장 오른쪽 아래 칸으로 규칙에 맞게 점프를 해서 가는 것이다.

각 칸에 적혀있는 수는 현재 칸에서 갈 수 있는 거리를 의미한다. 반드시 오른쪽이나 아래쪽으로만 이동해야 한다. 0은 더 이상 진행을 막는 종착점이며, 항상 현재 칸에 적혀있는 수만큼 오른쪽이나 아래로 가야 한다. 한 번 점프를 할 때, 방향을 바꾸면 안 된다. 즉, 한 칸에서 오른쪽으로 점프를 하거나, 아래로 점프를 하는 두 경우만 존재한다.

가장 왼쪽 위 칸에서 가장 오른쪽 아래 칸으로 규칙에 맞게 이동할 수 있는 경로의 개수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 게임 판의 크기 N (4 ≤ N ≤ 100)이 주어진다. 그 다음 N개 줄에는 각 칸에 적혀져 있는 수가 N개씩 주어진다. 칸에 적혀있는 수는 0보다 크거나 같고, 9보다 작거나 같은 정수이며, 가장 오른쪽 아래 칸에는 항상 0이 주어진다.

출력

가장 왼쪽 위 칸에서 가장 오른쪽 아래 칸으로 문제의 규칙에 맞게 갈 수 있는 경로의 개수를 출력한다. 경로의 개수는 2⁶³-1보다 작거나 같다.

Code

use std::io::{self, BufRead};

fn main() -> io::Result<()> {
    let stdin = io::stdin();
    let mut input = stdin.lock();
    let mut buffer = String::new();
    input.read_line(&mut buffer)?;
    let n: usize = buffer.trim().parse().unwrap();
    let mut graph: Vec<Vec<usize>> = Vec::new();
    for _ in 0..n {
        buffer.clear();
        input.read_line(&mut buffer)?;
        let row: Vec<usize> = buffer
            .split_whitespace()
            .map(|s| s.parse().unwrap())
            .collect();
        graph.push(row);
    }
    let mut dp: Vec<Vec<usize>> = vec![vec![0; n]; n];
    dp[0][0] = 1; // Initial value

    // Loop through the coordinates of the graph that can be reached
    for i in 0..n {
        for j in 0..n {
            // If the current coordinate is the bottom-right corner, stop the loop
            if i == n - 1 && j == n - 1 {
                println!("{}", dp[i][j]);
                return Ok(());
            }

            // Move to the right
            if j + graph[i][j] < n {
                dp[i][j + graph[i][j]] += dp[i][j];
            }

            // Move down
            if i + graph[i][j] < n {
                dp[i + graph[i][j]][j] += dp[i][j];
            }
        }
    }
    Ok(())
}

백준 1890 - 점프 [Rust]