結果

問題 No.1474 かさまＪ
ユーザー gorugo30
提出日時 2021-02-20 08:53:52
言語 Java
(openjdk 23)
結果
AC  
実行時間 1,693 ms / 2,500 ms
コード長 2,103 bytes
コンパイル時間 2,143 ms
コンパイル使用メモリ 77,604 KB
実行使用メモリ 351,184 KB
最終ジャッジ日時 2024-09-17 10:50:04
合計ジャッジ時間 15,470 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 17
権限があれば一括ダウンロードができます

ソースコード

diff # 

import java.util.*;

class Main{
    private static long MOD = 1000000007;
    private static long fac[] = new long[50000];
    private static long modpow(long x, long p){
        long ret = 1;
        while (p > 0){
            if (p % 2 == 1) ret = (ret * x) % MOD;
            x = (x * x) % MOD;
            p >>= 1;
        }
        return ret;
    }
    private static long nCr(int n, int r){
        if (n < r) return 0;
        return (((fac[n] * modpow(fac[n - r], MOD - 2)) % MOD) * modpow(fac[r], MOD - 2)) % MOD;
    }
    private static int min(int x, int y){
        if (x < y) return x;
        return y;
    }
    public static void main(String[] args){
        Scanner sc = new Scanner(System.in);
        int N = sc.nextInt();
        int MP = sc.nextInt();
        int Mq = sc.nextInt();
        int L = sc.nextInt();
        int S[] = new int[N];
        for (int i = 0; i < N; i++){
            S[i] = sc.nextInt();
        }
        
        long dp[][][] = new long[N][N + 1][Mq + 1];
        dp[0][0][Mq] = 1;
        for (int i = 1; i <= min(S[0], Mq); i++){
            dp[0][1][Mq - i] = 1;
        }

        for (int i = 1; i < N; i++){
            for (int k = 0; k <= N; k++){
                for (int r = 0; r <= Mq; r++){
                    dp[i][k][r] = dp[i - 1][k][r];
                }
            }
            for (int k = N; k >= 1; k--){
                long sum = 0;
                for (int r = Mq; r >= 0; r--){
                    dp[i][k][r] = (dp[i][k][r] + sum) % MOD;
                    sum = (sum + dp[i - 1][k - 1][r]) % MOD;
                    if (r + S[i] <= Mq) sum = (sum - dp[i - 1][k - 1][r + S[i]] + MOD) % MOD;
                }
            }
        }

        fac[0] = 1;
        for (int i = 1; i < 50000; i++){
            fac[i] = (fac[i - 1] * i) % MOD;
        }
        
        long ans = 0;
        for (int m = 0; m <= Mq; m++){
            for (int x = 0; x <= N; x++){
                ans = (ans + dp[N - 1][x][Mq - m] * nCr(MP + m - L * x + N - 1, N - 1)) % MOD;
            }
        }
        System.out.println(ans);
    }
}
0