結果

問題 No.1238 選抜クラス
ユーザー tcltk
提出日時 2021-01-17 05:15:05
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,169 ms / 2,000 ms
コード長 1,074 bytes
コンパイル時間 179 ms
コンパイル使用メモリ 82,344 KB
実行使用メモリ 309,256 KB
最終ジャッジ日時 2024-11-29 02:42:01
合計ジャッジ時間 16,682 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

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)]
    dp[0][0] = 1
    for n in range(N):
        dp1 = [[0 for _ in range(sumA+1)] for _ in range(N+1)]
        for k in range(n+1):
            for a in range(sumA+1):
                if dp[k][a] > 0:
                    dp1[k][a] += dp[k][a]
                    dp1[k+1][a+A[n]] += dp[k][a]
        dp = dp1

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

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