結果

問題 No.301 サイコロで確率問題 (1)
ユーザー cielciel
提出日時 2015-11-14 00:58:10
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 34 ms / 1,000 ms
コード長 1,761 bytes
コンパイル時間 382 ms
コンパイル使用メモリ 49,196 KB
実行使用メモリ 4,352 KB
最終ジャッジ日時 2023-10-11 17:33:47
合計ジャッジ時間 1,016 ms
ジャッジサーバーID
（参考情報） 		judge13 / judge14
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 34 ms
4,348 KB
testcase_01 AC 17 ms
4,352 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <vector>
#include <cstdio>
#include <cmath>
using namespace std;

//derived from checkio expected-dice
//unfortunately TLE using Python, rewriting in C++11.

bool gauss(vector<vector<double> > &a){
    if(a.empty())return false;
    int n=a.size();
    for(int i=0;i<n;i++){
        if(!a[i][i]){
            int j=i+1;
            for(;j<n;j++){
                if(a[j][i]){
                    for(int k=i;k<=n;k++)a[i][k]+=a[j][k];
                    break;
                }
            }
            if(j==n)return false;
        }
        for(int j=0;j<n;j++)if(i!=j){
            double r=a[j][i]/a[i][i];
            for(int k=i;k<=n;k++)a[j][k]=a[j][k]-a[i][k]*r;
        }
    }
    for(int i=0;i<n;i++){
        double x=a[i][i];
        for(int j=0;j<a[i].size();j++)a[i][j]/=x;
    }
    return true;
}

double expected(int n, int s, int t){//vector<int> b){
    //perform checkio probably-dice
    vector<double> a(s*(n+1)+1);
    for(int i=1;i<=s;i++)a[i+s]=pow(1.0/s,n);
    for(int e=0;e<n-1;e++)for(int i=s*n;i>=0;i--){
        double sum=0;
        for(int j=i;j<i+s;j++)sum+=a[j];
        a[i+s]=sum;
    }
    int l=t+1; //b.size();
    vector<vector<double> > A(l);
    for(int i=0;i<l;i++){
        A[i].resize(l+1);
        A[i][i]=1;
        if(i!=t)A[i][l]=1;
    }
    for(int start=0;start<l;start++)if(start!=t){
        for(int roll=n;roll<=s*n;roll++){
            int end=roll+start;
            //end = (end + b[end%l])%l;
            if(end>=l)end=0;
            A[start][end]-=a[roll+s];
        }
    }
    gauss(A);
    return A[0][A[0].size()-1];
}

int main(){
int T;
for(scanf("%d",&T);T--;){
    long long N;
    scanf("%lld",&N);
if(N>110)printf("%.15f\n",N+5.0/3);
else printf("%.15f\n",expected(1,6,N));
}
}
0