結果
| 問題 | No.1238 選抜クラス |
| コンテスト | |
| ユーザー |
|
| 提出日時 | 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 |
ソースコード
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)