結果

問題 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
権限があれば一括ダウンロードができます

ソースコード

diff # 

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)
0