[코드트리/INTERMEIDATE LOW] 사각형 채우기 3 (JAVA)

🔺 문제

코드트리 | 코딩테스트 준비를 위한 알고리즘 정석

 

 

🔺 코드

import java.util.*;
import java.io.*;
 
public class Main {
    static final int mod = 1_000_000_007;
 
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
 
        long[] dp = new long[1001];
        dp[0] = 1;  // 전혀 놓지 않는 경우 1가지
        dp[1] = 2;
 
        for(int i = 2 ; i <= N ; i++) {
            dp[i] = (dp[i - 1] * 2 + dp[i - 2] * 3) % mod;
 
            for(int j = i - 3 ; j >= 0 ; j--) {
                dp[i] = (dp[i] + dp[j] * 2) % mod;
            }
        }
 
        System.out.println(dp[N]);
    }
}

 

 

 

🧩  해결 아이디어

• DP - subproblem을 그대로 합치면 되는 DP

- (i-1)번째까지 채워져있고, 나머지 한 열만 채우는 경우 => 2가지 가능

- (i-2)번째까지 채워져있고, 나머지 두 열을 채우는 경우 => 3가지 가능

- (i-k)번째까지 채워져있고, 남은 k열을 채우는 경우 (단, k는 3보다 크고 i보다 작음) => 2가지 가능

dp[i] = 2 * dp[i - 1] + 3 * dp[i - 2] + 2 * (dp[i - 3] + dp[i - 4] + ... + dp[1] + dp[0]) (단, i ≥ 2)

 

 

💥 유의사항

- DP 배열, long형 배열로 선언해야 함!

 

 

---

💬 느낀 점

만만하지 않은 dp....

오래간만입니다..

 

1회독	2회독	3회독	4회독	5회독
V

---

(참고)

✔ 코드트리