結果

問題 No.2653 [Cherry 6th Tune] Re: start! (Make it Zero!)
ユーザー coindarwcoindarw
提出日時 2024-03-01 20:46:59
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 88 ms / 2,000 ms
コード長 1,219 bytes
コンパイル時間 3,203 ms
コンパイル使用メモリ 254,228 KB
実行使用メモリ 6,548 KB
最終ジャッジ日時 2024-03-01 20:47:12
合計ジャッジ時間 11,836 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,548 KB
testcase_01 AC 37 ms
6,548 KB
testcase_02 AC 42 ms
6,548 KB
testcase_03 AC 41 ms
6,548 KB
testcase_04 AC 48 ms
6,548 KB
testcase_05 AC 47 ms
6,548 KB
testcase_06 AC 86 ms
6,548 KB
testcase_07 AC 88 ms
6,548 KB
testcase_08 AC 51 ms
6,548 KB
testcase_09 AC 49 ms
6,548 KB
testcase_10 AC 53 ms
6,548 KB
testcase_11 AC 47 ms
6,548 KB
testcase_12 AC 51 ms
6,548 KB
testcase_13 AC 32 ms
6,548 KB
testcase_14 AC 32 ms
6,548 KB
testcase_15 AC 32 ms
6,548 KB
testcase_16 AC 32 ms
6,548 KB
testcase_17 AC 33 ms
6,548 KB
testcase_18 AC 32 ms
6,548 KB
testcase_19 AC 32 ms
6,548 KB
testcase_20 AC 32 ms
6,548 KB
testcase_21 AC 31 ms
6,548 KB
testcase_22 AC 32 ms
6,548 KB
testcase_23 AC 34 ms
6,548 KB
testcase_24 AC 32 ms
6,548 KB
testcase_25 AC 33 ms
6,548 KB
testcase_26 AC 34 ms
6,548 KB
testcase_27 AC 35 ms
6,548 KB
testcase_28 AC 36 ms
6,548 KB
testcase_29 AC 34 ms
6,548 KB
testcase_30 AC 33 ms
6,548 KB
testcase_31 AC 34 ms
6,548 KB
testcase_32 AC 35 ms
6,548 KB
testcase_33 AC 34 ms
6,548 KB
testcase_34 AC 40 ms
6,548 KB
testcase_35 AC 38 ms
6,548 KB
testcase_36 AC 36 ms
6,548 KB
testcase_37 AC 41 ms
6,548 KB
testcase_38 AC 40 ms
6,548 KB
testcase_39 AC 37 ms
6,548 KB
testcase_40 AC 37 ms
6,548 KB
testcase_41 AC 37 ms
6,548 KB
testcase_42 AC 34 ms
6,548 KB
testcase_43 AC 85 ms
6,548 KB
testcase_44 AC 86 ms
6,548 KB
testcase_45 AC 84 ms
6,548 KB
testcase_46 AC 85 ms
6,548 KB
testcase_47 AC 85 ms
6,548 KB
testcase_48 AC 84 ms
6,548 KB
testcase_49 AC 88 ms
6,548 KB
testcase_50 AC 85 ms
6,548 KB
testcase_51 AC 84 ms
6,548 KB
testcase_52 AC 84 ms
6,548 KB
testcase_53 AC 34 ms
6,548 KB
testcase_54 AC 40 ms
6,548 KB
testcase_55 AC 43 ms
6,548 KB
testcase_56 AC 31 ms
6,548 KB
testcase_57 AC 36 ms
6,548 KB
testcase_58 AC 34 ms
6,548 KB
testcase_59 AC 42 ms
6,548 KB
testcase_60 AC 37 ms
6,548 KB
testcase_61 AC 44 ms
6,548 KB
testcase_62 AC 41 ms
6,548 KB
testcase_63 AC 28 ms
6,548 KB
testcase_64 AC 28 ms
6,548 KB
testcase_65 AC 29 ms
6,548 KB
testcase_66 AC 29 ms
6,548 KB
testcase_67 AC 29 ms
6,548 KB
testcase_68 AC 30 ms
6,548 KB
testcase_69 AC 29 ms
6,548 KB
testcase_70 AC 29 ms
6,548 KB
testcase_71 AC 29 ms
6,548 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

#include <atcoder/modint.hpp>
#define reps(i, n) for (int i = 1, i##_len = (n); i <= i##_len; ++i)
using namespace std;
using namespace atcoder;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    int Q;
    cin >> Q;
    while (Q--) {
        int n, m;
        cin >> n >> m;
        using mint = modint;
        mint::set_mod(m);
        using mint998 = modint998244353;

        vector<mint> a(n + 1);
        vector<int> b(n + 1);
        reps(i, n) {
            int x;
            cin >> x;
            if (x != -1) a[i] = x;
            else b[i] = 1;
        }
        auto ca = a;
        auto cb = b;
        reps(i, n) ca[i] += ca[i - 1];
        reps(i, n) cb[i] += cb[i - 1];

        mint998 ans = ((cb[n] == 0) ? (ca[n] == 0) : mint998(m).pow(cb[n] - 1)) * (n - 1);
        reps(i, n - 1) {
            mint lSum = ca[i], rSum = ca[n] - lSum;
            int lCnt = cb[i], rCnt = cb[n] - lCnt;
            mint998 lNum = (lCnt == 0) ? (lSum == 0) : mint998(m).pow(lCnt - 1);
            mint998 rNum = (rCnt == 0) ? (rSum == 0) : mint998(m).pow(rCnt - 1);
            ans -= lNum * rNum;
        }
        cout << ans.val() << endl;
    }

    return 0;
}
0