알고리즘

백준 17271번 리그 오브 레전설 (JAVA)

박카스마시며코딩 2023. 3. 8. 13:32

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

 

17271번: 리그 오브 레전설 (Small)

규환이는 리그 오브 레전설이라는 게임을 좋아한다. 이 게임에서는 N초의 시간 동안 싸움을 하는데, 규환이가 플레이하는 캐릭터는 A, B 두 가지 스킬을 사용할 수 있다.  A 스킬의 시전 시간은 1

www.acmicpc.net

 

 

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

public class Main {
    private static final int LIMIT = 1_000_000_007;
    private static final int NOT_VALID = -1;
    public static void main(String[] args) throws Exception {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st = new StringTokenizer(br.readLine());
        Function<String,Integer> stoi = Integer::parseInt;
        int n = stoi.apply(st.nextToken());
        int m = stoi.apply(st.nextToken());
        int[] dp = new int[n+1];
        Arrays.fill(dp,NOT_VALID);
        int result = cal(n,m,dp);
        System.out.println(result);
    }

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