알고리즘

백준 8394번 악수 (JAVA)

박카스마시며코딩 2022. 9. 22. 17:42

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

 

8394번: 악수

첫째 줄에 회의에 참석한 사람의 수 n (1 ≤ n ≤ 10,000,000)이 주어진다.

www.acmicpc.net

 

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

DP를 이용해 이전에 계산한 값을 다시 계산하지 않도록 하였습니다.

n이 오른쪽 즉, n-1번째 사람과 악수를 할지 안할지를 계산하였습니다.

n이 1이하라면 옆사람이 존재하지 않기에 이때는 악수하는 것을 고려하지 않았습니다.

n이 0이면 끝까지 온것이기에 1을 리턴하도록 하였습니다.

 

package BOJ.dp;

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

public class BOJ_8394 {

    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+1];
        int result = cal(n,dp);
        System.out.println(result);
    }

    private static int cal(int n,int[] dp) {
        if(n == 0){
            return 1;
        }
        if(dp[n] != 0){
            return dp[n];
        }
        int result = 0;
        if(n > 1){
            result += cal(n-2,dp);
        }
        result += cal(n-1,dp);
        result %= 10;
        dp[n] = result;
        return result;
    }
}