結果

問題 No.2940 Sigma Sigma Div Floor Problem
ユーザー rlangevin
提出日時 2024-10-21 08:54:03
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 222 ms / 6,000 ms
コード長 573 bytes
コンパイル時間 549 ms
コンパイル使用メモリ 82,556 KB
実行使用メモリ 75,812 KB
最終ジャッジ日時 2024-10-21 08:54:14
合計ジャッジ時間 11,584 ms
ジャッジサーバーID
(参考情報)
judge1 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 29 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #

# https://qiita.com/AkariLuminous/items/3e2c80baa6d5e6f3abe9
# \sum_{i=0}^{n-1} (a * i + b)//m
def floor_sum(n, m, a, b):
    ans = 0
    while True:
        if a >= m or a < 0:
            ans += n * (n - 1) * (a // m) // 2
            a %= m
        if b >= m or b < 0:
            ans += n * (b // m)
            b %= m
        y_max = a * n + b
        if y_max < m: break
        n, b, m, a = y_max // m, y_max % m, a, m
    return ans


N = int(input())
mod = 998244353
ans = 0
for i in range(1, N + 1):
    ans += floor_sum(N + 1, i, 1, 0)
    ans %= mod

print(ans)
0