結果

問題 No.301 サイコロで確率問題 (1)
ユーザー fumofumofunifumofumofuni
提出日時 2021-09-08 18:03:07
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 394 ms / 1,000 ms
コード長 2,346 bytes
コンパイル時間 2,220 ms
コンパイル使用メモリ 208,820 KB
実行使用メモリ 6,940 KB
最終ジャッジ日時 2024-06-07 20:48:09
合計ジャッジ時間 3,493 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

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

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
#define rep(i,n) for(ll i=0;i<n;i++)
#define repl(i,l,r) for(ll i=(l);i<(r);i++)
#define per(i,n) for(ll i=(n)-1;i>=0;i--)
#define perl(i,r,l) for(ll i=r-1;i>=l;i--)
#define fi first
#define se second
#define pb push_back
#define ins insert
#define pqueue(x) priority_queue<x,vector<x>,greater<x>>
#define all(x) (x).begin(),(x).end()
#define CST(x) cout<<fixed<<setprecision(x)
#define rev(x) reverse(x);
using ll=long long;
using vl=vector<ll>;
using vvl=vector<vector<ll>>;
using pl=pair<ll,ll>;
using vpl=vector<pl>;
using vvpl=vector<vpl>;
const ll MOD=1000000007;
const ll MOD9=998244353;
const int inf=2e9+1;
const ll INF=4e18;
const ll dy[8]={-1,0,1,0,1,1,-1,-1};
const ll dx[8]={0,-1,0,1,1,-1,1,-1};
template <typename T> inline bool chmax(T &a, T b) {
  return ((a < b) ? (a = b, true) : (false));
}
template <typename T> inline bool chmin(T &a, T b) {
  return ((a > b) ? (a = b, true) : (false));
}

using ld=long double;
ll mod=1000000007;//一次元ベクトルの高速化dp
vector<ld> matmul(vector<ld> &dp, vector<vector<ld>> &mt){
    ll m=dp.size();
    vector<ld> ret(m,0); 
    rep(i,m)rep(j,m){
      ret[i]+=mt[i][j]*dp[j];
    }
    return ret;
}
vector<vector<ld>> update(vector<vector<ld>> &mt){
    ll m=mt.size();
    vector<vector<ld>> ret(m,vector<ld>(m,0));
    rep(i,m)rep(j,m)rep(k,m){
        ret[i][j]+=mt[i][k]*mt[k][j];
    }
    return ret;
}
void matpow(vector<ld> &dp, vector<vector<ld>> &mt, ll k){
    ll m=dp.size();
    while(k){
        if(k&1)dp=matmul(dp,mt);
        mt=update(mt);
        k/=2;
    }
}

ld modpow(ld a,ll n) {
  ld res = 1;
  while (n > 0) {
      if (n & 1) res = res * a;
      a = a * a;
      n >>= 1;
  }
  return res;
}

void solve(){
    ll n;cin >> n;
    if(n>=20000){
        ld ans=n;ans+=(ld)5/3;
        cout << ans << endl;
        return;
    }
    vector<ld> dp(6,1);
    dp[0]=0;
    vector<vector<ld>> mat(6,vector<ld>(6));
    rep(i,6)mat[0][i]=(ld)1.0/6;
    rep(i,5)mat[i+1][i]=1;
    matpow(dp,mat,n);
    vector<ld> ndp(7,0);ndp[6]=1;
    vector<vector<ld>> nmat(7,vector<ld>(7));
    rep(i,6)nmat[0][i]=(ld)1.0/6;nmat[0][6]=1;
    rep(i,5)nmat[i+1][i]=1;nmat[6][6]=1;
    matpow(ndp,nmat,n);
    cout << ndp[0]/((ld)1.0-dp[0]) << endl;
}
int main(){
    ll t;cin >> t;
    CST(20);
    while(t--)solve();
}
0