結果
問題 |
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)