結果
問題 | No.269 見栄っ張りの募金活動 |
ユーザー |
|
提出日時 | 2021-07-22 14:10:54 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 124 ms / 5,000 ms |
コード長 | 877 bytes |
コンパイル時間 | 195 ms |
コンパイル使用メモリ | 82,224 KB |
実行使用メモリ | 91,188 KB |
最終ジャッジ日時 | 2024-07-17 14:43:41 |
合計ジャッジ時間 | 2,270 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 22 |
ソースコード
from sys import exitn, s, k = map(int, input().split())#x[i]:= (i 人目の生徒の支払う金額) - i*K とすると、#x[0] + x[1] + ⋯+x[N-1] = S−(K+2K+⋯+(N-1)K)#0 <= x[0] <= x[1] <= ... <= x[N-1]#を満たす (x[0], x[1], ..., x[N-1]) の組を数え上げる問題になるので、これは分割数 P(S−(K+2K+⋯+(N-1)K), N) です。mod = 10**9 + 7S = s - k*(n-1)*n//2def part_num(n, k):if n == k == 0:return 1mod = 10**9 + 7dp = [[0] * (k + 1) for _ in range(n + 1)]for j in range(1, k+1):dp[0][j] = 1for i in range(1, n+1):for j in range(1, k+1):if i - j >= 0:dp[i][j] = (dp[i][j-1] + dp[i-j][j])%modelse:dp[i][j] = dp[i][j-1]return dp[n][k]if S < 0:print(0)exit()ans = part_num(S, n)print(ans%mod)