結果

問題 No.3462 Buttons
コンテスト
ユーザー InTheBloom
提出日時 2026-02-28 15:43:02
言語 D
(dmd 2.112.0)
コンパイル:
dmd -fPIE -m64 -w -wi -O -release -inline -I/opt/dmd/src/druntime/import/ -I/opt/dmd/src/phobos -L-L/opt/dmd/linux/lib64/ -fPIC _filename_
実行:
./Main
結果
AC  
実行時間 377 ms / 2,000 ms
コード長 2,481 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 4,666 ms
コンパイル使用メモリ 191,864 KB
実行使用メモリ 7,844 KB
最終ジャッジ日時 2026-02-28 15:43:14
合計ジャッジ時間 9,727 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1
other AC * 15
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

import std;

void main () {
    const long MOD = 998244353;
    int caseNum = readln.chomp.to!int;
    auto ans = new long[](caseNum);
    foreach (caseId; 0 .. caseNum) {
        int A, B, K;
        readln.read(A, B, K);

        if (B == 0 || B == 1) {
            if (0 <= A) {
                ans[caseId] = 1L * A * K % MOD;
            }
            else {
                ans[caseId] = 0;
            }

            continue;
        }

        if (B == -1) {
            if (0 <= A) {
                ans[caseId] = 1L * A * K % MOD;
            }
            else {
                ans[caseId] = 1L * -A * (K - 1) % MOD;
            }
            continue;
        }

        if (2 <= B) {
            if (0 <= A) {
                ans[caseId] = A * mod_pow(B, K - 1, MOD) % MOD;
            }
            else {
                ans[caseId] = 0;
            }
            continue;
        }

        if (B <= -2) {
            long rem = K;
            if (0 <= A) {
                long v = A;
                rem--;

                if (rem % 2 == 1) {
                    v += A;
                    rem--;
                }

                v *= mod_pow(-B, rem, MOD);
                v %= MOD;
                ans[caseId] = v;
            }
            else {
                if (rem == 1) {
                    ans[caseId] = 0;
                }
                else {
                    long v = -A;
                    rem--;

                    if (rem % 2 == 0) {
                        v += -A;
                        rem--;
                    }

                    v *= mod_pow(-B, rem, MOD);
                    v %= MOD;
                    ans[caseId] = v;
                }
            }
            continue;
        }

        enforce(caseId < 0);
    }

    writefln("%(%s\n%)", ans);
}

void read (T...) (string S, ref T args) {
    import std.conv : to;
    import std.array : split;
    auto buf = S.split;
    foreach (i, ref arg; args) {
        arg = buf[i].to!(typeof(arg));
    }
}

long mod_pow (long a, long x, const long MOD)
in {
    assert(0 <= x, "x must satisfy 0 <= x");
    assert(1 <= MOD, "MOD must satisfy 1 <= MOD");
    assert(MOD <= int.max, "MOD must satisfy MOD*MOD <= long.max");
}
do {
    // normalize
    a %= MOD; a += MOD; a %= MOD;

    long res = 1L;
    long base = a;
    while (0 < x) {
        if (0 < (x&1)) (res *= base) %= MOD;
        (base *= base) %= MOD;
        x >>= 1;
    }

    return res % MOD;
}
0