백준 14651번 걷다보니 신천역 삼 (JAVA)

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

14651번: 걷다보니 신천역 삼 (Large)

 

저는 DP를 이용하여 문제를 해결하였습니다.

3의 배수는 각 자리 수의 합이 3의 배수이면 3의 배수입니다.

dfs를 각 자리수의 합만을 인자로 보내고 뎁스가 n이고 3의 배수면 1을 리턴하도록 하였습니다.

dp는 뎁스에 대한 합입니다. 하지만 합은 3의 배수인지만 판별하기 위해 쓰이기 때문에 합%3에 해당하는 가지 수를 저장하였습니다.

 

package BOJ.dp;

import java.io.BufferedReader;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.function.Function;

public class BOJ_14651 {
    private static final int LIMIT = 1_000_000_000 + 9;
    private static final int NOT_VALID = -1;
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Function<String,Integer> stoi = Integer::parseInt;
        int n = stoi.apply(br.readLine());
        int[][] dp = new int[n][3];
        for(int i = 0 ; i < n;  i++){
            Arrays.fill(dp[i],NOT_VALID);
        }
        int result = cal(0,0,n,dp);
        System.out.println(result);
    }

    private static int cal(int depth,int sum, int n,int[][] dp) {
        if(depth == n && sum % 3 == 0){
            return 1;
        }
        if(depth == n){
            return 0;
        }
        if(dp[depth][sum] != NOT_VALID){
            return dp[depth][sum];
        }
        int result = 0;
        for(int i = 0 ; i < 3; i++){
            if(depth == 0 && i == 0){
                continue;
            }
            result += cal(depth+1,(sum + i) % 3,n,dp) % LIMIT;
            result %= LIMIT;
        }
        dp[depth][sum] = result;
        return result;
    }
}