結果

問題 No.1238 選抜クラス
ユーザー tcltk
提出日時 2021-01-17 05:13:39
言語 PyPy3
(7.3.15)
結果
MLE  
実行時間 -
コード長 1,041 bytes
コンパイル時間 261 ms
コンパイル使用メモリ 82,068 KB
実行使用メモリ 849,144 KB
最終ジャッジ日時 2024-11-29 02:41:00
合計ジャッジ時間 23,253 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 27 MLE * 9
権限があれば一括ダウンロードができます

ソースコード

diff # 

#region Header
#!/usr/bin/env python3
# from typing import *

import sys
import io
import math
import collections
import decimal
import itertools
from queue import PriorityQueue
import bisect
import heapq

def input():
    return sys.stdin.readline()[:-1]

sys.setrecursionlimit(1000000)
#endregion

# _INPUT = """2 2
# 1 3
# """
# sys.stdin = io.StringIO(_INPUT)

MOD = 1000000007

def main():
    N, K = map(int, input().split())
    A = list(map(int, input().split()))
    sumA = sum(A)

    dp = [[[0 for _ in range(sumA+1)] for _ in range(N+1)] for _ in range(N+1)]
    dp[0][0][0] = 1
    for n in range(N):
        for k in range(n+1):
            for a in range(sumA+1):
                if dp[n][k][a] > 0:
                    dp[n+1][k][a] += dp[n][k][a]
                    dp[n+1][k+1][a+A[n]] += dp[n][k][a]
    
    v = 0
    for k in range(1, N+1):
        for a in range(K*k, sumA+1):
            v = (v + dp[N][k][a]) % MOD
    print(v)

if __name__ == '__main__':
    main()
0