N×N 게임판에 수가 적혀져 있다. 이 게임의 목표는 가장 왼쪽 위 칸에서 가장 오른쪽 아래 칸으로 규칙에 맞게 점프를 해서 가는 것이다.
각 칸에 적혀있는 수는 현재 칸에서 갈 수 있는 거리를 의미한다. 반드시 오른쪽이나 아래쪽으로만 이동해야 한다. 0은 더 이상 진행을 막는 종착점이며, 항상 현재 칸에 적혀있는 수만큼 오른쪽이나 아래로 가야 한다. 한 번 점프를 할 때, 방향을 바꾸면 안 된다. 즉, 한 칸에서 오른쪽으로 점프를 하거나, 아래로 점프를 하는 두 경우만 존재한다.
가장 왼쪽 위 칸에서 가장 오른쪽 아래 칸으로 규칙에 맞게 이동할 수 있는 경로의 개수를 구하는 프로그램을 작성하시오.
첫째 줄에 게임 판의 크기 N (4 ≤ N ≤ 100)이 주어진다. 그 다음 N개 줄에는 각 칸에 적혀져 있는 수가 N개씩 주어진다. 칸에 적혀있는 수는 0보다 크거나 같고, 9보다 작거나 같은 정수이며, 가장 오른쪽 아래 칸에는 항상 0이 주어진다.
가장 왼쪽 위 칸에서 가장 오른쪽 아래 칸으로 문제의 규칙에 맞게 갈 수 있는 경로의 개수를 출력한다. 경로의 개수는 263-1보다 작거나 같다.
4
2 3 3 1
1 2 1 3
1 2 3 1
3 1 1 0
3
그리 어려운 문제는 아니다. DP를 통해 시간 초과가 나지 않게 처리하면 바로 통과되겠거니 생각했지만 놓친 부분이 있어 시간이 조금 더 걸렸다.
#include <iostream>
#include <cstring>
#include <vector>
using namespace std;
int arr[101][101];
int n;
long long cache[101][101];
long long jump(int row, int col){
// 범위 벗어날 경우 0 리턴
if (row >= n || col>= n){
cache[row][col] = 0;
return 0;
}
// 도착지에 도착한 경우 1 리턴
else if (row == n-1 && col == n-1){
cache[row][col] = 1;
return 1;
}
long long& result = cache[row][col];
// 캐시에 해당 값이 미리 저장된 경우 그 값을 리턴
if (result != -1) return cache[row][col];
// 만약 arr[row][col] 값이 0이면 더이상 진행을 못하므로 0리턴해야함.
if (arr[row][col] == 0) return result = 0;
// 한 지점에서 두가지 경로에 대한 값을 재귀적으로 더함
return result = jump(row+arr[row][col], col) + jump(row, col+arr[row][col]);
}
int main(){
cin>>n;
for (int i=0; i<n; i++){
for (int j=0; j<n; j++){
cin>>arr[i][j];
}
}
memset(cache, -1, sizeof(cache));
long long test = jump(0,0);
cout<<test;
}
0으로 둘러싸일 경우 더이상 진행하지 못하기에 바로 리턴시키고 종료해야 함을 놓쳤다.
memset은 cstring 헤더를 필요로 한다는 것도 유의