結果

問題 No.1238 選抜クラス
ユーザー persimmon-persimmon
提出日時 2021-03-03 10:28:42
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 61 ms / 2,000 ms
コード長 794 bytes
コンパイル時間 187 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 62,848 KB
最終ジャッジ日時 2024-10-03 03:09:14
合計ジャッジ時間 3,479 ms
ジャッジサーバーID
(参考情報)
judge2 / judge5
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff #

def main1(n,k,a):
  mod=10**9+7
  aa=[x-k for x in a]
  am=[]
  a0=[]
  ap=[]
  for x in aa:
    if x==0:a0.append(x)
    if x<0:am.append(x)
    if x>0:ap.append(x)
  dpp=[0]*(101*n)
  dpm=[0]*(101*n)
  dpp[0]=1
  for x in ap:
    for i in range(101*n-1,-1,-1):
      if dpp[i]==0:continue
      dpp[i+x]+=dpp[i]
      dpp[i+x]%=mod
  dpm[0]=1
  for x in am:
    x=abs(x)
    for i in range(101*n-1,-1,-1):
      if dpm[i]==0:continue
      dpm[i+x]+=dpm[i]
      dpm[i+x]%=mod
  sdpp=[0]
  for x in dpp:sdpp.append(sdpp[-1]+x)
  ret=0
  for i in range(101*n):
    ret+=dpm[i]*(sdpp[-1]-sdpp[i])
    ret%=mod
  ret*=pow(2,len(a0),mod)%mod
  ret-=1
  return ret%mod


if __name__=='__main__':
  n,k=map(int,input().split())
  a=list(map(int,input().split()))
  ret1=main1(n,k,a)
  print(ret1)
0