結果

問題 No.3462 Buttons
コンテスト
ユーザー wasd314
提出日時 2026-02-28 16:08:53
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 456 ms / 2,000 ms
コード長 1,038 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 200 ms
コンパイル使用メモリ 77,820 KB
実行使用メモリ 76,228 KB
最終ジャッジ日時 2026-02-28 16:09:03
合計ジャッジ時間 8,109 ms
ジャッジサーバーID
(参考情報)
judge7 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 1
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code

from sys import stdin
def input():
    return stdin.readline().rstrip("\n")
mod = 998244353

def solve(a, b, k):
    if a == 0:
        return 0
    if k == 1:
        return max(a, 0)
    if b == 0 or b == 1:
        if a <= 0:
            return 0
        else:
            return a * k % mod
    if b > 1:
        if a <= 0:
            return 0
        # 1 * a + (k-1) * b
        return a * pow(b, k - 1, mod) % mod
    if b == -1:
        if a > 0:
            return a * k % mod
        else:
            return -a * (k - 1) % mod

    kh = (k - 1) // 2
    b2 = b * b
    if a > 0:
        if k % 2 == 0:
            return a * 2 * pow(b2, kh, mod) % mod
        else:
            return a * pow(b2, kh, mod) % mod
    if k % 2 == 0:
        return a * pow(b, k - 1, mod) % mod
    else:
        return a * 2 * pow(b, k - 2, mod) % mod

if __name__ == "__main__":
    case_t = 1
    case_t = int(input())
    for _ in [None] * case_t:
        a, b, k = map(lambda s_: int(s_), input().split())
        print(solve(a, b, k) % mod)
0