結果
問題 | No.1238 選抜クラス |
ユーザー |
|
提出日時 | 2025-02-20 16:03:32 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 991 ms / 2,000 ms |
コード長 | 2,308 bytes |
コンパイル時間 | 268 ms |
コンパイル使用メモリ | 82,196 KB |
実行使用メモリ | 93,960 KB |
最終ジャッジ日時 | 2025-02-20 16:03:47 |
合計ジャッジ時間 | 13,389 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 36 |
ソースコード
import os,sys,random,threading #sys.exit() 退出程序 #sys.setrecursionlimit(10**6) #调整栈空间 from random import randint,choice,shuffle #randint(a,b)从[a,b]范围随机选择一个数 #choice(seq)seq可以是一个列表,元组或字符串,从seq中随机选取一个元素 #shuffle(x)将一个可变的序列x中的元素打乱 from copy import deepcopy from io import BytesIO,IOBase from types import GeneratorType from functools import lru_cache,reduce #reduce(op,迭代对象) from bisect import bisect_left,bisect_right #bisect_left(x) 大于等于x的第一个下标 #bisect_right(x) 大于x的第一个下标 from collections import Counter,defaultdict,deque from itertools import accumulate,combinations,permutations #accumulate(a)用a序列生成一个累积迭代器,一般list化前面放个[0]做前缀和用 #combinations(a,k)a序列选k个 组合迭代器 #permutations(a,k)a序列选k个 排列迭代器 from heapq import heapify,heappop,heappush #heapify将列表转为堆 from typing import Generic,Iterable,Iterator,TypeVar,Union,List from string import ascii_lowercase,ascii_uppercase,digits #小写字母,大写字母,十进制数字 from math import ceil,floor,sqrt,pi,factorial,gcd,log,log10,log2,inf #ceil向上取整,floor向下取整 ,sqrt开方 ,factorial阶乘 from decimal import Decimal,getcontext #Decimal(s) 实例化Decimal对象,一般使用字符串 #getcontext().prec=100 修改精度 from sys import stdin, stdout, setrecursionlimit input = lambda: sys.stdin.readline().rstrip("\r\n") MI = lambda :map(int,input().split()) li = lambda :list(MI()) ii = lambda :int(input()) mod = int(1e9 + 7) #998244353 inf = 1<<60 py = lambda :print("YES") pn = lambda :print("NO") DIRS = [(0, 1), (1, 0), (0, -1), (-1, 0)] # 右下左上 DIRS8 = [(0, 1), (1, 1), (1, 0), (1, -1), (0, -1), (-1, -1), (-1, 0),(-1, 1)] # →↘↓↙←↖↑↗ n,k=li() a=li() tot=sum(a) f=[[0]*(tot+1) for _ in range(n+1)] f[0][0]=1 for x in a: for i in range(n-1,-1,-1): for j in range(tot+1): if j+x<=tot: f[i+1][j+x]+=f[i][j] f[i+1][j+x]%=mod res=0 for i in range(1,n+1): for j in range(tot,i*k-1,-1): res+=f[i][j] # if f[i][j]: # print(i,j,f[i][j]) res%=mod print(res)