結果
問題 | No.2940 Sigma Sigma Div Floor Problem |
ユーザー |
![]() |
提出日時 | 2024-10-18 21:57:41 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 96 ms / 6,000 ms |
コード長 | 635 bytes |
コンパイル時間 | 332 ms |
コンパイル使用メモリ | 82,788 KB |
実行使用メモリ | 64,004 KB |
最終ジャッジ日時 | 2024-10-18 23:34:53 |
合計ジャッジ時間 | 3,261 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 29 WA * 1 |
ソースコード
mod = 998244353 def floor_sum(n: int, m: int, a: int, b: int) -> int: assert 1 <= n assert 1 <= m ans = 0 if a >= m: ans += (n - 1) * n * (a // m) // 2 a %= m if b >= m: ans += n * (b // m) b %= m y_max = (a * n + b) // m x_max = y_max * m - b if y_max == 0: return ans ans += (n - (x_max + a - 1) // a) * y_max % mod ans += floor_sum(y_max, a, m, (a - x_max % a) % a) % mod return ans N = int(input()) if N > 1e7: exit() ans = 0 for j in range(1, N+1): ans += floor_sum(N+1, j, 1, 0) - floor_sum(j, j, 1, 0) ans %= mod print(ans)