PS

[백준] 17484번 진우의 달 여행 (Small)

Eastplanet 2022. 6. 24. 21:48

문제 출처: https://www.acmicpc.net/problem/17484

 

17484번: 진우의 달 여행 (Small)

첫줄에 지구와 달 사이 공간을 나타내는 행렬의 크기를 나타내는 N, M (2≤ N, M ≤ 6)이 주어진다. 다음 N줄 동안 각 행렬의 원소 값이 주어진다. 각 행렬의 원소값은 100 이하의 자연수이다.

www.acmicpc.net

moon(n,m,내가 다음에 움직일 방향)으로 재귀함수를 작성한다.

위 함수는 moon(n-1,m + 내가 이전에 움직인 방향, 내가 이전에 움직인 방향)을 호출한다.

이때 2번의 같은 방향으로 움직이는 것은 불가능하기 때문에 내가 다음에 움직일 방향 != 내가 이전에 움직인 방향이다.

Small에서는 메모이제이션을 사용하지 않아도 통과한다.

#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include <string>

using namespace std;

int N, M;
int arr[1001][1001];
int dp[1001][1001][3];
int movePos[3] = { -1,0,1 };
const int INF = 123456789;

int moon(int n, int m, int dir) {
    if (n == -1)return 0;
    if (dp[n][m][dir] != INF)return dp[n][m][dir];
    
    int minV = INF;
    for (int i = 0; i < 3; i++) {
        if (i == dir)continue;
        if (m + movePos[i] < 0 || m + movePos[i] >= M)continue;

        minV = min(minV, moon(n - 1, m + movePos[i], i) + arr[n][m]);
    }

    dp[n][m][dir] = minV;
    
    return minV;
}


int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    cout.tie(NULL);

    cin >> N >> M;

    for (int i = 0; i < N; i++) {
        for (int j = 0; j < M; j++) {
            for (int k = 0; k < 3; k++) {
                dp[i][j][k] = INF;
            }
        }
    }

  
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < M; j++) {
            cin >> arr[i][j];
        }
    }

    int minV = INF;
    
    for (int j = 0; j < M; j++) {
        for (int i = 0; i < 3; i++) {
            minV = min(minV,moon(N-1, j, i));
        }
    }

    cout << minV;
    
    

}