結果

問題 No.2001 Distanced Triple
コンテスト
ユーザー lam6er
提出日時 2025-03-26 15:52:47
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 40 ms / 2,000 ms
コード長 1,779 bytes
コンパイル時間 492 ms
コンパイル使用メモリ 82,780 KB
実行使用メモリ 54,268 KB
最終ジャッジ日時 2025-03-26 15:53:14
合計ジャッジ時間 2,661 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 30
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

MOD = 998244353
inv2 = pow(2, MOD-2, MOD)
inv6 = pow(6, MOD-2, MOD)

def sum_x(a, b, MOD, inv2):
    if a > b:
        return 0
    a_mod = a % MOD
    b_mod = b % MOD
    sum_b = (b_mod * (b_mod + 1)) % MOD
    sum_b = (sum_b * inv2) % MOD
    a_minus_1 = (a - 1) % MOD
    sum_a = (a_minus_1 * a_mod) % MOD
    sum_a = (sum_a * inv2) % MOD
    return (sum_b - sum_a) % MOD

def sum_x2(a, b, MOD, inv6):
    if a > b:
        return 0
    a_mod = a % MOD
    b_mod = b % MOD
    sum_b = (b_mod * (b_mod + 1)) % MOD
    sum_b = (sum_b * ((2 * b_mod + 1) % MOD)) % MOD
    sum_b = (sum_b * inv6) % MOD
    a_minus_1 = (a - 1) % MOD
    sum_a = (a_minus_1 * a_mod) % MOD
    sum_a = (sum_a * ((2 * a_mod - 1) % MOD)) % MOD
    sum_a = (sum_a * inv6) % MOD
    return (sum_b - sum_a) % MOD

L, R = map(int, input().split())
A, B, C = map(int, input().split())

X_max = min(R - A, R - C, R - (A + B))
if X_max < L:
    print(0)
else:
    if (C - B - A) >= 0:
        a = inv2
        term_b_numerator = (2 * (A + B - R) - 3) % MOD
        b = (term_b_numerator * inv2) % MOD
        term_c1 = ((C - B - A) % MOD) * ((R - C + 1) % MOD) % MOD
        term_c2_part1 = (R + 1 - C) % MOD
        term_c2_part2 = (R + 2 - C) % MOD
        term_c2 = (term_c2_part1 * term_c2_part2) % MOD
        term_c2 = (term_c2 * inv2) % MOD
        c = (term_c1 + term_c2) % MOD
    else:
        K = (R - B - A + 1) % MOD
        a = inv2
        term_b_numerator = (- (2 * K + 1)) % MOD
        b = (term_b_numerator * inv2) % MOD
        c_part = (K * (K + 1)) % MOD
        c = (c_part * inv2) % MOD

    sum_x2_val = sum_x2(L, X_max, MOD, inv6)
    sum_x_val = sum_x(L, X_max, MOD, inv2)
    cnt = (X_max - L + 1) % MOD
    total = (a * sum_x2_val + b * sum_x_val + c * cnt) % MOD
    print(total)
0