結果

問題 No.3505 Sum of Prod of Root
コンテスト
ユーザー kitatai
提出日時 2026-04-19 00:30:59
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
TLE  
実行時間 -
コード長 1,640 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 208 ms
コンパイル使用メモリ 85,376 KB
実行使用メモリ 193,456 KB
最終ジャッジ日時 2026-04-19 00:31:37
合計ジャッジ時間 7,828 ms
ジャッジサーバーID
（参考情報） 		judge3_1 / judge1_0
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 10 RE * 2 TLE * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import math

N = int(input())
M = 998244353


def h(n):
    if n <= 0:
        return 0
    m = math.isqrt(n)
    return (
        ((m - 1) * m * (2 * m - 1) * (3 * m * m - 3 * m - 1)) // 15 % M
        + 3 * (((m - 1) * m) // 2) ** 2 % M
        + ((m - 1) * m * (2 * m - 1)) // 6 % M
        + m * ((n * (n + 1)) // 2 % M - ((m * m - 1) * (m * m)) // 2) % M
    ) % M


next_u = [1] * 60
next_ui = [1] * 60
for k in range(3, 60):
    next_ui[k] += 1
    next_u[k] = next_ui[k] ** k
u_time = [0] * 1000000
u_change = [None] * 1000000
u_time[0] = 1
ut_len = 1
v = 1
while v <= N:
    min_u = next_u[3]
    for k in range(3, 60):
        if min_u > next_u[k]:
            min_u = next_u[k]
    ks = []
    for k in range(3, 60):
        if next_u[k] == min_u:
            ks.append(k)
    for k in ks:
        next_ui[k] += 1
        next_u[k] = next_ui[k] ** k
    if u_time[ut_len - 1] != min_u:
        u_time[ut_len] = min_u
        u_change[ut_len - 1] = ks
        ut_len += 1
    v = min_u
ans = 0
f = False

max_c = int(N ** (1 / 3)) + 10
while (max_c + 1) ** 3 <= N:
    max_c += 1
while max_c**3 > N:
    max_c -= 1
max_c += 10

inv = [0] * (max_c + 1)
inv[1] = 1
for i in range(2, max_c + 1):
    inv[i] = M - (M // i) * inv[M % i] % M
cu = [1] * 60
p = 1

for i in range(ut_len - 1):
    u_i = u_time[i]
    u_i1 = u_time[i + 1]
    if u_i1 > N:
        u_i1 = N + 1
        f = True
    seg = (h(u_i1 - 1) - h(u_i - 1)) % M
    ans = (ans + p * seg) % M

    if f:
        break
    for k in u_change[i]:
        old = cu[k]
        cu[k] += 1
        new = cu[k]
        p = p * inv[old] % M
        p = p * new % M
print(ans)
0