쉬운 프로그래밍

[알고리즘] 백준 2193 이친수 본문

알고리즘/DP

[알고리즘] 백준 2193 이친수

쉬운형 2020. 4. 26. 18:53

 

import java.util.Scanner;

public class BJ_2193 {
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int n = scan.nextInt();
        long[][] dp = new long[n + 1][2];

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

        if (n >= 2) {
            dp[2][0] = 1;
            dp[2][1] = 0;

            if (n >= 3) {
                for (int i = 3; i <= n; i++) {
                    dp[i][1] = dp[i - 1][1] + dp[i - 2][1];
                    dp[i][0] = dp[i - 1][0] * 2 - dp[i - 1][1];
                }
            }
        }

        System.out.println(dp[n][0]);
    }
}

 

전형적인 dp 문제

 

dp[n - 1]의 2배값에서 dp[n - 1]에 들어갈 이친수 중 맨 끝값이 1인 경우의수를 빼주면 정답을 구할 수 있다.

 

dp[n - 1]에서 맨 끝값이 1인 경우의 수의 점화식을 구하는데 좀 오래걸렸다.

 

맞은거같은데 계속 틀려서 뭐지했는데 n이 커지면 integer 범위를 초과하므로 dp배열을 long으로 바꿔줬다...

Comments