結果

問題 No.2874 Gunegune Tree
ユーザー 寝癖寝癖
提出日時 2024-07-14 18:00:07
言語 C++23
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 1,071 ms / 2,000 ms
コード長 1,991 bytes
コンパイル時間 3,549 ms
コンパイル使用メモリ 261,456 KB
実行使用メモリ 14,156 KB
最終ジャッジ日時 2024-07-14 18:05:48
合計ジャッジ時間 18,839 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,816 KB
testcase_01 AC 2 ms
6,816 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 75 ms
6,940 KB
testcase_04 AC 434 ms
7,808 KB
testcase_05 AC 157 ms
6,940 KB
testcase_06 AC 3 ms
6,944 KB
testcase_07 AC 1,008 ms
13,492 KB
testcase_08 AC 467 ms
8,068 KB
testcase_09 AC 45 ms
6,940 KB
testcase_10 AC 851 ms
11,992 KB
testcase_11 AC 658 ms
10,076 KB
testcase_12 AC 204 ms
6,944 KB
testcase_13 AC 975 ms
13,444 KB
testcase_14 AC 808 ms
11,648 KB
testcase_15 AC 136 ms
6,944 KB
testcase_16 AC 1,042 ms
13,828 KB
testcase_17 AC 115 ms
6,944 KB
testcase_18 AC 125 ms
6,944 KB
testcase_19 AC 84 ms
6,940 KB
testcase_20 AC 110 ms
6,940 KB
testcase_21 AC 331 ms
6,940 KB
testcase_22 AC 492 ms
8,448 KB
testcase_23 AC 310 ms
6,940 KB
testcase_24 AC 410 ms
7,424 KB
testcase_25 AC 703 ms
10,696 KB
testcase_26 AC 432 ms
7,680 KB
testcase_27 AC 1,023 ms
13,604 KB
testcase_28 AC 229 ms
6,940 KB
testcase_29 AC 585 ms
9,344 KB
testcase_30 AC 463 ms
8,192 KB
testcase_31 AC 343 ms
6,944 KB
testcase_32 AC 1,071 ms
14,156 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace std;
using namespace atcoder;

using mint = modint998244353;
const mint zero = 0, one = 1;

// 状態i->jに遷移する確率
const vector<vector<mint>> P = {
    {zero, one/5, one/5, one/5, one/5, one/5},
    {zero, one/2, one/2, zero , zero , zero },
    {zero, one/3, one/3, one/3, zero , zero },
    {zero, zero , one/3, one/3, one/3, zero },
    {zero, zero , zero , one/3, one/3, one/3},
    {zero, zero , zero , zero , one/2, one/2}
};

// 単位行列
const vector<vector<mint>> E = {
    {one, zero, zero, zero, zero, zero},
    {zero, one, zero, zero, zero, zero},
    {zero, zero, one, zero, zero, zero},
    {zero, zero, zero, one, zero, zero},
    {zero, zero, zero, zero, one, zero},
    {zero, zero, zero, zero, zero, one}
};

// 行列積
vector<vector<mint>> operator*(const vector<vector<mint>> &A, const vector<vector<mint>> &B) {
    int N = A.size();
    vector<vector<mint>> C(N, vector<mint>(N, 0));
    for (int i = 0; i < N; i++) {
        for (int j = 0; j < N; j++) {
            for (int k = 0; k < N; k++) {
                C[i][j] += A[i][k]*B[k][j];
            }
        }
    }
    return C;
}

int main() {
    int N;
    cin >> N;

    // 0: 初期状態, 1: 右, 2: 右上, 3: 上, 4: 左上, 5: 左
    vector<vector<mint>> dp(N+1, vector<mint>(6));

    // P^i, P^{i+1}
    vector<vector<mint>> Pi = E, Pi1 = P;

    for (int i = 0; i < N; i++) {
        for (int j = 1; j < 6; j++) {
            for (int k = 0; k < 6; k++) {
                // (i+1)日目の状態がjであるという条件のもとで、i日目の状態がkである確率
                mint pk = P[k][j]*Pi[0][k]/Pi1[0][j];
                // 1<j<5ならば高さが1増える
                dp[i+1][j] += (dp[i][k]+(1<j&&j<5))*pk;
            }
        }
        Pi = Pi1;
        Pi1 = Pi*P;
    }

    mint ans = 0;
    for (int i = 0; i < 6; i++) ans += dp[N][i]*Pi[0][i];
    cout << ans.val() << endl;
}
0