[Java | DP | 2342번 | 골3] Dance Dance Revolution

☑️ 문제

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

 

승환이는 요즘 "Dance Dance Revolution"이라는 게임에 빠져 살고 있다. 하지만 그의 춤 솜씨를 보면 알 수 있듯이, 그는 DDR을 잘 하지 못한다. 그럼에도 불구하고 그는 살을 뺄 수 있다는 일념으로 DDR을 즐긴다.

DDR은 아래의 그림과 같은 모양의 발판이 있고, 주어진 스텝에 맞춰 나가는 게임이다. 발판은 하나의 중점을 기준으로 위, 아래, 왼쪽, 오른쪽으로 연결되어 있다. 편의상 중점을 0, 위를 1, 왼쪽을 2, 아래를 3, 오른쪽을 4라고 정하자.

처음에 게이머는 두 발을 중앙에 모으고 있다.(그림에서 0의 위치) 그리고 게임이 시작하면, 지시에 따라 왼쪽 또는 오른쪽 발을 움직인다. 하지만 그의 두 발이 동시에 움직이지는 않는다.

이 게임에는 이상한 규칙이 더 있다. 두 발이 같은 지점에 있는 것이 허락되지 않는 것이다. (물론 게임 시작시에는 예외이다) 만약, 한 발이 1의 위치에 있고, 다른 한 발이 3의 위치에 있을 때, 3을 연속으로 눌러야 한다면, 3의 위치에 있는 발로 반복해야 눌러야 한다는 것이다.

오랫동안 DDR을 해 온 백승환은 발이 움직이는 위치에 따라서 드는 힘이 다르다는 것을 알게 되었다. 만약, 중앙에 있던 발이 다른 지점으로 움직일 때, 2의 힘을 사용하게 된다. 그리고 다른 지점에서 인접한 지점으로 움직일 때는 3의 힘을 사용하게 된다. (예를 들면 왼쪽에서 위나 아래로 이동할 때의 이야기이다.) 그리고 반대편으로 움직일때는 4의 힘을 사용하게 된다. (위쪽에서 아래쪽으로, 또는 오른쪽에서 왼쪽으로). 만약 같은 지점을 한번 더 누른다면, 그때는 1의 힘을 사용하게 된다.

만약 1 → 2 → 2 → 4를 눌러야 한다고 가정해 보자. 당신의 두 발은 처음에 (point 0, point 0)에 위치하여 있을 것이다. 그리고 (0, 0) → (0, 1) → (2, 1) → (2, 1) → (2, 4)로 이동하면, 당신은 8의 힘을 사용하게 된다. 다른 방법으로 발을 움직이려고 해도, 당신은 8의 힘보다 더 적게 힘을 사용해서 1 → 2 → 2 → 4를 누를 수는 없을 것이다.

 

입력

입력은 지시 사항으로 이루어진다. 각각의 지시 사항은 하나의 수열로 이루어진다. 각각의 수열은 1, 2, 3, 4의 숫자들로 이루어지고, 이 숫자들은 각각의 방향을 나타낸다. 그리고 0은 수열의 마지막을 의미한다. 즉, 입력 파일의 마지막에는 0이 입력된다. 입력되는 수열의 길이는 100,000을 넘지 않는다.

 

출력

한 줄에 모든 지시 사항을 만족하는 데 사용되는 최소의 힘을 출력한다.

 

예제 입력1

1 2 2 4 0

 

예제 출력1

8

 

☄️ 정답코드

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

public class Main {
    static int INF = Integer.MAX_VALUE;
    static List<Integer> steps = new ArrayList<>();;
    static int[][][] dp;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        StringTokenizer st = new StringTokenizer(br.readLine());
        while(true) {
            int n = Integer.parseInt(st.nextToken());
            if(n == 0) break;
            steps.add(n);
        }

        int N = steps.size();

        dp = new int[N+1][5][5];
        for (int i = 0; i < N+1; i++) {
            for (int j = 0; j < 5; j++) {
                Arrays.fill(dp[i][j], INF);
            }
        }

        dp[0][0][0] = 0;

        for (int i = 0; i < N; i++) {
            int move = steps.get(i);
            for (int left = 0; left < 5; left++) {
                for (int right = 0; right < 5; right++) {
                    if(dp[i][left][right] == INF) continue;

                    if (right != move) {
                        int power = dp[i][left][right] + getCost(left, move);
                        dp[i + 1][move][right] = Math.min(dp[i + 1][move][right], power);
                    }

                    if (left != move) {
                        int power = dp[i][left][right] + getCost(right, move);
                        dp[i + 1][left][move] = Math.min(dp[i + 1][left][move], power);
                    }
                }
            }
        }

        int result = INF;
        for (int left = 0; left < 5; left++) {
            for (int right = 0; right < 5; right++) {
                result = Math.min(result, dp[N][left][right]);
            }
        }

        System.out.println(result);
    }

    static int getCost(int from, int to) {
        if(from == 0) return 2;
        if(from == to) return 1;
        if((from == 1 && to == 3) || (from == 3 && to == 1) ||
                (from == 2 && to == 4) || (from == 4 && to == 2)) return 4;
        return 3;
    }
}

🪐 풀이

DP

 

어려워서 스스로 풀지는 못했던 문제 ...........

3중 배열 dp로 푸는 문제는 아직 너무 어렵다

😿

 

✅ dp

여기서 수상하게 봐야할 포인트가 있다

각 발이 0~5까지 놓일 수 있다는 것!

5칸이라는 점이 배열로 만들어도 크지 않은 적당한 크기의 숫자이다

 

그래서 dp를 아래와 같은 의미로 만들 수 있다

dp[step][left][right]: 변수명 그대로, 주어진 step에 왼발(left), 오른발(right)이 어디있는지 표현

 

[초기화]

dp 모든 칸을 INF로 초기화 시켜주고

처음 위치인 dp[0][0][0]은 힘이 들지 않으니까 0으로 해준다!

 

[이동]

입력값으로 주어진 steps를 순서대로 현재 위치에서 이동하면서 dp 테이블 갱신

// 왼발 움직이는 경우
if (right != move) {
    int power = dp[i][left][right] + getCost(left, move);
    dp[i+1][move][right] = Math.min(dp[i+1][move][right], power);
}


// 오른발 움직이는 경우
if (left != move) {
    int power = dp[i][left][right] + getCost(right, move);
    dp[i+1][left][move] = Math.min(dp[i+1][left][move], power);
}

 

[비용 계산]

문제에 설명한대로 계산해주면 된다!

static int getCost(int from, int to) {
	// 1. 발이 중앙에서 이동: 힘2
    if(from == 0) return 2;
    
    // 2. 발이 같은 지점 힘1
    if(from == to) return 1;
    
    // 3. 발이 반대편으로 이동: 힘4
    if((from == 1 && to == 3) || (from == 3 && to == 1) ||
       (from == 2 && to == 4) || (from == 4 && to == 2)) return 4;
    
    // 4. 인접한 지점으로 이동: 힘3
    return 3;
}