結果

問題 No.2891 Mint
ユーザー LyricalMaestroLyricalMaestro
提出日時 2024-10-06 23:06:36
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 138 ms / 2,000 ms
コード長 1,895 bytes
コンパイル時間 154 ms
コンパイル使用メモリ 82,484 KB
実行使用メモリ 61,824 KB
最終ジャッジ日時 2024-10-06 23:06:43
合計ジャッジ時間 5,743 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 38 ms
52,096 KB
testcase_01 AC 37 ms
52,864 KB
testcase_02 AC 42 ms
58,624 KB
testcase_03 AC 37 ms
52,480 KB
testcase_04 AC 57 ms
58,112 KB
testcase_05 AC 37 ms
52,480 KB
testcase_06 AC 135 ms
61,696 KB
testcase_07 AC 38 ms
51,968 KB
testcase_08 AC 41 ms
52,480 KB
testcase_09 AC 37 ms
52,352 KB
testcase_10 AC 36 ms
52,480 KB
testcase_11 AC 38 ms
52,352 KB
testcase_12 AC 37 ms
52,352 KB
testcase_13 AC 37 ms
52,608 KB
testcase_14 AC 36 ms
52,352 KB
testcase_15 AC 37 ms
52,352 KB
testcase_16 AC 37 ms
52,352 KB
testcase_17 AC 111 ms
61,696 KB
testcase_18 AC 93 ms
61,568 KB
testcase_19 AC 80 ms
61,184 KB
testcase_20 AC 105 ms
61,696 KB
testcase_21 AC 82 ms
61,696 KB
testcase_22 AC 118 ms
61,696 KB
testcase_23 AC 74 ms
61,568 KB
testcase_24 AC 104 ms
61,184 KB
testcase_25 AC 126 ms
61,440 KB
testcase_26 AC 121 ms
61,440 KB
testcase_27 AC 118 ms
61,696 KB
testcase_28 AC 138 ms
61,824 KB
testcase_29 AC 82 ms
61,568 KB
testcase_30 AC 107 ms
61,696 KB
testcase_31 AC 86 ms
61,440 KB
testcase_32 AC 113 ms
61,696 KB
testcase_33 AC 94 ms
61,184 KB
testcase_34 AC 98 ms
61,440 KB
testcase_35 AC 104 ms
61,440 KB
testcase_36 AC 109 ms
61,824 KB
testcase_37 AC 54 ms
57,344 KB
testcase_38 AC 54 ms
57,856 KB
testcase_39 AC 62 ms
57,984 KB
testcase_40 AC 59 ms
57,600 KB
testcase_41 AC 50 ms
58,368 KB
testcase_42 AC 37 ms
52,796 KB
testcase_43 AC 37 ms
52,096 KB
testcase_44 AC 38 ms
52,820 KB
testcase_45 AC 37 ms
52,480 KB
testcase_46 AC 39 ms
52,736 KB
testcase_47 AC 56 ms
58,112 KB
testcase_48 AC 56 ms
58,112 KB
testcase_49 AC 58 ms
57,600 KB
testcase_50 AC 45 ms
57,984 KB
testcase_51 AC 52 ms
57,984 KB
testcase_52 AC 44 ms
57,600 KB
testcase_53 AC 50 ms
57,600 KB
testcase_54 AC 51 ms
57,728 KB
testcase_55 AC 42 ms
57,984 KB
testcase_56 AC 58 ms
58,112 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

## https://yukicoder.me/problems/no/1385

import math

MOD = 998244353

inv2 = pow(2, MOD - 2, MOD)

def sum_k(n):
    ans = (n * (n + 1)) % MOD
    ans *= inv2
    ans %= MOD
    return ans

def solve_normal(N, M):
    answer = 0
    for k in range(1, N + 1):
        n = (M % k)
        answer += n
        answer %= MOD
    return answer

def solve_true(N, M):
    sqrt_n = int(math.sqrt(M))

    # 前半は正規に解く
    answer1 = 0
    for k in range(1, min(N, sqrt_n) + 1):
        n = (M % k)
        answer1 += n
        answer1 %= MOD

    # 後半は商に注目
    start = N
    answer2 = 0
    for q in range(sqrt_n + 1):
        end = M // (q + 1)
        end = max(sqrt_n, end)
        if start > end:
            # [start, end)が商qの範囲
            if q == 0:
                n1 = (N - M) % MOD
                ans = (n1 * M) % MOD
                answer2 += ans
                answer2 %= MOD
            else:
                init_s = M // q
                init_r = M % q

                a1 = (-q * sum_k(start)) % MOD
                b1 = (init_s * q) % MOD
                b1 += init_r
                b1 %= MOD
                b1 *= start
                b1 %= MOD

                a2 = (-q * sum_k(end)) % MOD
                b2 = (init_s * q) % MOD
                b2 += init_r
                b2 %= MOD
                b2 *= end
                b2 %= MOD

                ans = (a1 -a2) % MOD
                ans += b1
                ans %= MOD
                ans -= b2
                ans %= MOD

                answer2 += ans
                answer2 %= MOD
            start = end
        if end == sqrt_n:
            break

    return (answer1 + answer2) % MOD




def main():

    N, M = map(int, input().split())

#    ans1 = solve_normal(N, M)
    ans2 = solve_true(N, M)

#    print(ans1)
    print(ans2)



if __name__ == "__main__":
    main()
0