結果

問題 No.3462 Buttons
コンテスト
ユーザー tassei903
提出日時 2026-02-28 14:30:13
言語 PyPy3
(7.3.17)
コンパイル:
pypy3 -mpy_compile _filename_
実行:
pypy3 _filename_
結果
AC  
実行時間 420 ms / 2,000 ms
コード長 2,518 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 261 ms
コンパイル使用メモリ 78,396 KB
実行使用メモリ 75,292 KB
最終ジャッジ日時 2026-02-28 14:30:27
合計ジャッジ時間 8,116 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import sys
input = lambda :sys.stdin.readline()[:-1]
ni = lambda :int(input())
na = lambda :list(map(int,input().split()))
yes = lambda :print("yes");Yes = lambda :print("Yes");YES = lambda : print("YES")
no = lambda :print("no");No = lambda :print("No");NO = lambda : print("NO")
#######################################################################

def naive(a, b, K):
    ans = 0
    for k in range(K + 1):
        for i in range(1 << k):
            z = 0
            for j in range(k):
                if i >> j & 1:
                    z += a
                else:
                    z *= b
            # print(k, bin(i), z)
            ans = max(ans, z)
    return ans

def test(a, b, k):
    if a == 0:
        return 0
    elif b == 0:
        return max(0, a * k)
    elif b == 1:
        return max(0, a * k)
    elif b > 1:
        if a <= 0:
            return 0
        else:
            return a * b ** (k - 1)
    elif b == -1:
        if a > 0:
            return a * k
        else:
            return -a * (k - 1)
    else:
        if a > 0:
            if k % 2 == 1:
                return a * b ** (k-1)
            else:
                return 2 * a * b ** (k-2)
        else:
            if k % 2 == 0:
                return a * b ** (k-1)
            elif k == 1:
                return 0
            else:
                return 2 * a * b ** (k-2)

mod = 998244353
def solve(a, b, k):
    if a == 0:
        return 0
    elif b == 0 or b == 1:
        if a > 0:
            return a * k % mod
        else:
            return 0
    elif b > 1:
        if a < 0:
            return 0
        else:
            return a * pow(b, k-1, mod) % mod
    elif b == -1:
        if a > 0:
            return a * k % mod
        else:
            return (-a * (k - 1) )% mod
    else:
        if a > 0:
            if k % 2 == 1:
                return a * pow(b, k-1, mod) % mod
            else:
                return 2 * a * pow(b, k-2, mod) % mod
        else:
            if k % 2 == 0:
                return a * pow(b, k-1, mod) % mod
            elif k == 1:
                return 0
            else:
                return 2 * a * pow(b, k-2, mod) % mod


# from random import randint
# for _ in range(100000):
#     k = randint(1, 3)
#     a = randint(-5, 5)
#     b = randint(-5, 5)
#     r1 = naive(a, b, k)
#     r2 = test(a, b, k)
#     if r1 != r2:
#         print(a, b, k)
#         print(r1, r2)
#         break

for _ in range(ni()):
    a, b, k = na()
    print(solve(a, b, k))
0