#1149622 (PyPy3) No.3462 Buttons

• 問題: No.3462 Buttons
• ユーザー: tyawanmusi
• 提出日時: 2026-02-27 10:23:49
• 言語: PyPy3 (7.3.17)
• 結果: AC
• 実行時間: 821 ms / 2,000 ms
• コード長: 5,160 bytes
• コンパイル時間: 410 ms
• コンパイル使用メモリ: 77,884 KB
• 実行使用メモリ: 85,920 KB
• 最終ジャッジ日時: 2026-02-28 13:13:27
• 合計ジャッジ時間: 14,086 ms
• ジャッジサーバーID （参考情報）: judge4 / judge7

ファイルパターン	結果
sample	AC * 1
other	AC * 15

ソースコード

import sys

class InputError(Exception):
    pass

class StrictIn:

    def __init__(self, s: bytes, i: int = 0, line: int = 1, col: int = 1):
        self.s = s
        self.i = i
        self.line = line
        self.col = col

    @staticmethod
    def from_stdin_ascii() -> "StrictIn":
        raw = sys.stdin.buffer.read()
        for p, b in enumerate(raw):
            if b >= 0x80:
                raise InputError(f"non-ASCII byte at offset {p}: 0x{b:02x}")
            if b in (0x09, 0x0A, 0x0D):
                continue
            if 0x20 <= b <= 0x7E:
                continue
            raise InputError(f"disallowed control byte at offset {p}: 0x{b:02x}")
        return StrictIn(raw)

    def _eof(self) -> bool:
        return self.i >= len(self.s)

    def _peek(self):
        return None if self._eof() else self.s[self.i]

    def _advance(self) -> int:
        if self._eof():
            self._err("unexpected EOF")
        b = self.s[self.i]
        self.i += 1
        if b == 0x0A:
            self.line += 1
            self.col = 1
        else:
            self.col += 1
        return b

    def _err(self, msg: str) -> None:
        raise InputError(f"{msg} (line {self.line}, col {self.col})")

    def _consume_newline(self) -> None:
        b = self._peek()
        if b == 0x0A:
            self._advance()
            return
        if b == 0x0D:
            self._advance()
            if self._peek() != 0x0A:
                self._err("bare CR is not allowed (use LF or CRLF)")
            self._advance()
            self.line += 1
            self.col = 1
            return
        self._err("expected EOL")

    def skip_spaces(self) -> None:
        while (not self._eof()) and (self._peek() in (0x20, 0x09)):
            self._advance()

    def skip_ws(self) -> None:
        while not self._eof():
            b = self._peek()
            if b in (0x20, 0x09):
                self._advance()
            elif b in (0x0A, 0x0D):
                self._consume_newline()
            else:
                break

    def read_space(self) -> None:
        b = self._peek()
        if b not in (0x20, 0x09):
            self._err("expected SPACE/TAB")
        while (not self._eof()) and (self._peek() in (0x20, 0x09)):
            self._advance()

    def read_eoln(self) -> None:
        self._consume_newline()

    def read_eof(self) -> None:
        if not self._eof():
            self._err("expected EOF (extra data exists)")

    def read_token(self) -> str:
        b = self._peek()
        if b is None:
            self._err(f"unexpected EOF while reading token")
        if b in (0x20, 0x09):
            self._err(f"unexpected leading SPACE/TAB while reading token")
        if b in (0x0A, 0x0D):
            self._err(f"unexpected EOL while reading token")

        start = self.i
        while not self._eof():
            b = self._peek()
            if b in (0x20, 0x09, 0x0A, 0x0D):
                break
            self._advance()
        return self.s[start:self.i].decode("ascii")

    def read_int(self, lo=None, hi=None) -> int:
        t_str = self.read_token()
        t = t_str.encode("ascii")

        if t[:1] == b"-":
            body = t[1:]
            if len(body) == 0:
                self._err(f"int is not an integer")
        else:
            body = t

        if len(body) == 0 or (not all(48 <= c <= 57 for c in body)):
            self._err(f"int is not a base-10 integer")

        x = int(t_str)
        if lo is not None and x < lo:
            self._err(f"int out of range: {x} < {lo}")
        if hi is not None and x > hi:
            self._err(f"int out of range: {x} > {hi}")
        return x

    def read_string(self, min_len: int | None = None, max_len: int | None = None) -> str:

        s = self.read_token()

        if min_len is not None and len(s) < min_len:
            self._err(f"str too short: len={len(s)} < {min_len}")
        if max_len is not None and len(s) > max_len:
            self._err(f"str too long: len={len(s)} > {max_len}")

        return s

ins = StrictIn.from_stdin_ascii()
t = ins.read_int(lo = 1, hi = 2*10**5)
ins.read_eoln()
for _ in range(t):
    a = ins.read_int(lo = -10**9, hi = 10**9)
    ins.read_space()
    b = ins.read_int(lo = -10**9, hi = 10**9)
    ins.read_space()
    k = ins.read_int(lo = 1, hi = 10**9)
    ins.read_eoln()
    mod = 998244353
    if b == 1 or b == 0:
        print(max(a, 0) * k % mod)
        continue
    if b == -1:
        if a >= 0:
            print(a * k % mod)
        else:
            print(- a * (k - 1) % mod)
        continue
    if a < 0:
        if b > 0:
            print(0)
        else:
            if k == 1:
                print(0)
            else:
                if k % 2 == 0:
                    print(- a * pow(-b, k-1, mod) % mod)
                else:
                    print(- 2 * a * pow(-b, k-2, mod) % mod)
    else:
        if b > 0:
            print(a * pow(b, k-1, mod) % mod)
        else:
            if k % 2 == 0:
                print(2 * a * pow(-b, k-2, mod) % mod)
            else:
                print(a * pow(-b, k-1, mod) % mod)
ins.read_eof()