結果

問題 No.75 回数の期待値の問題
ユーザー ooaiu
提出日時 2025-09-14 02:56:45
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 14 ms / 5,000 ms
コード長 1,219 bytes
コンパイル時間 2,985 ms
コンパイル使用メモリ 284,300 KB
実行使用メモリ 7,716 KB
最終ジャッジ日時 2025-09-14 02:56:49
合計ジャッジ時間 4,094 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 16
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#include <atcoder/modint>
using namespace std;
using mint = atcoder::modint998244353;
using ld = long double;
int main() {
    std::ios_base::sync_with_stdio(false);
    std::cin.tie(nullptr);
    int K;
    cin >> K;
    vector<vector<ld>> A(K + 1, vector<ld>(K + 2));
    for (int i = 0; i < K; i++) {
        A[i][i] = 6;
        for (int j = 1; j <= 6; j++) {
            int k = (i + j > K ? 0 : i + j);
            A[i][k] -= 1;
        }
        A[i][K + 1] = 6;
    }
    A[K][K] = 1;
    int rank = 0;
    for (int j = 0; j < K + 2; j++) {
        for (int i = rank; i < K + 1; i++) {
            if (A[i][j] != 0) {
                if (i != rank) swap(A[i], A[rank]);
                break;
            }
        }
        if (A[rank][j] == 0) continue;
        ld t = A[rank][j];
        for (int it = 0; it < K + 2; it++) A[rank][it] /= t;
        for (int i = 0; i < K + 1; i++) {
            if (i != rank && A[i][j] != 0) {
                ld x = A[i][j];
                for (int it = 0; it < K + 2; it++) A[i][it] -= x * A[rank][it];
            }
        }
        rank++;
        if (rank == K + 1) break;
    }
    cout << fixed << setprecision(16) << A[0][K + 1] << "\n";
}
0