結果
問題 |
No.1474 かさまJ
|
ユーザー |
![]() |
提出日時 | 2021-04-09 21:55:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,362 ms / 2,500 ms |
コード長 | 935 bytes |
コンパイル時間 | 278 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 360,904 KB |
最終ジャッジ日時 | 2024-06-25 05:14:06 |
合計ジャッジ時間 | 8,999 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 17 |
ソースコード
N,P,Q,L=map(int,input().split()) S=list(map(int,input().split())) mod=10**9+7 def cmb(n,r): if r<0 or r>n: return 0 return (g1[n]*g2[r]*g2[n-r])%mod g1=[1,1] g2=[1,1] inv=[0,1] for i in range(2,50003): g1.append((g1[-1]*i)%mod) inv.append((-inv[mod%i]*(mod//i))%mod) g2.append((g2[-1]*inv[-1])%mod) DP=[[[0]*(Q+2) for j in range(i+2)] for i in range(N+1)] DP[0][0][0]=1 C=[[[0]*(Q+2) for j in range(i+2)] for i in range(N+1)] for i in range(N+1): for j in range(i+1): for k in range(Q+1): C[i][j][k]%=mod DP[i][j][k]+=C[i][j][k] DP[i][j][k]%=mod C[i][j][k+1]+=C[i][j][k] if i==N: break for j in range(i+1): for k in range(Q+1): DP[i+1][j+1][k]+=DP[i][j][k] C[i+1][j][k+1]+=DP[i][j][k] C[i+1][j][min(Q+1,k+S[i]+1)]-=DP[i][j][k] ANS=0 for i in range(N+1): for j in range(Q+1): x=L*(N-i)-j if x<=P: ANS+=cmb(N+P-x-1,P-x)*DP[N][i][j] print(ANS%mod)