본문 바로가기
Problem Solving/Baekjoon

[백준] 1793 타일링- Dynamic Programming / Java

by graycode 2023. 3. 26.

 문제 링크

 

1793번: 타일링

입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스는 한 줄로 이루어져 있으며, 정수 n이 주어진다.

www.acmicpc.net

 

 풀이 과정

문제의 타일링 방법을 배열을 활용한 점화식으로 나타내면 dp[n] = dp[n - 1] + dp[n - 2] * 2 가 성립한다.

다만 long 범위를 벗어나는 문제의 출력범위를 표현하기 위해서는 Java 에선 BigInteger 를 활용한다.

 

n의 범위가 최대 250 이므로, 모든 경우의 수를 구하고 입력받은 타일 길이의 경우의 수를 도출할 수 있지만,

해당 풀이에서는 입력 중 최대 값을 먼저 구해 해당 범위까지만 경우의 수를 구하였다.

 

 풀이 코드

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.math.BigInteger;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

public class Main {

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

        List<Integer> list = new ArrayList<>();
        String input;

        while ((input = br.readLine()) != null)
            list.add(Integer.parseInt(input));

        BigInteger[] dp = new BigInteger[Collections.max(list) + 1];

        if (dp.length > 1)
            dp[0] = dp[1] = BigInteger.valueOf(1);

        for (int i = 2; i < dp.length; i++)
            dp[i] = dp[i - 1].add(dp[i - 2].multiply(BigInteger.valueOf(2)));

        StringBuilder sb = new StringBuilder();
        for (int i : list)
            sb.append(dp[i]).append("\n");

        bw.write(sb.toString());
        bw.flush();
    }

}

댓글