結果

問題 No.76 回数の期待値で練習
ユーザー nebukuro09nebukuro09
提出日時 2016-12-19 22:30:52
言語 D
(dmd 2.106.1)
結果
AC  
実行時間 1,917 ms / 5,000 ms
コード長 2,113 bytes
コンパイル時間 2,947 ms
コンパイル使用メモリ 156,800 KB
実行使用メモリ 47,980 KB
最終ジャッジ日時 2024-06-12 05:56:37
合計ジャッジ時間 5,130 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 1,917 ms
47,980 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import std.stdio, std.array, std.string, std.conv, std.algorithm;
import std.typecons, std.range, std.random, std.math, std.container;
import std.numeric, std.bigint, core.bitop, core.stdc.stdio;


int gaussianElimination(ref real[][] org_mat, ref real[] ret) {
    int N = org_mat.length.to!int;

    auto mat = new real[][](N, N+1);
    foreach (i; 0..N)
        foreach (j; 0..N+1)
            mat[i][j] = org_mat[i][j];

    foreach (i; 0..N) {
        real maxval = 0;
        int maxrow = -1;
        foreach (j; i..N)
            if (abs(mat[j][i]) > maxval)
                maxval = abs(mat[j][i]), maxrow = j;
        if (maxval == 0)
            return -1;
        swap(mat[i], mat[maxrow]);

        foreach (j; i+1..N+1)
            mat[i][j] /= mat[i][i];
        mat[i][i] = 1;

        foreach (j; 0..N) {
            if (j == i) continue;
            real mul = mat[j][i];
            foreach (k; i..N+1)
                mat[j][k] -= mul*mat[i][k];
        }
    }

    foreach (i; 0..N)
        ret[i] = mat[i][N];

    return 0;
}


void main() {
    real[6] E = [1.0000000000000000,
                 1.0833333333333333,
                 1.2569444444444444,
                 1.5353009259259260,
                 1.6915991512345676,
                 2.0513639724794235];

    auto mat = new real[][](6, 7);
    foreach (i; 0..6) foreach(j; 0..7) mat[i][j] = 0;
    foreach (i; 0..6) {
        foreach (j; 1..7) {
            if (j >= i + 1)
                mat[i][j-1] += 1;
            else
                mat[i][j-1] += E[i-j]+1;
        }
        mat[i][6] = E[i];
    }

    auto P = new real[](6);
    gaussianElimination(mat, P);

    auto T = readln.chomp.to!int;
    while (T--) {
        auto N = readln.chomp.to!int;
        auto dp = new real[](N+1);
        foreach (i; 0..N+1)
            dp[i] = 0;
        foreach (i; iota(N-1, -1, -1)) {
            foreach (j; 1..7) {
                if (i + j >= N)
                    dp[i] += P[j-1];
                else
                    dp[i] += P[j-1] * (dp[i+j] + 1);
            }
        }
        writefln("%.10f", dp[0]);
    }
}
0