結果

問題 No.108 トリプルカードコンプ
ユーザー batsumaru
提出日時 2019-09-04 21:14:17
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 14 ms / 5,000 ms
コード長 1,191 bytes
コンパイル時間 1,860 ms
コンパイル使用メモリ 175,808 KB
実行使用メモリ 11,904 KB
最終ジャッジ日時 2024-12-29 17:11:59
合計ジャッジ時間 3,008 ms
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
#define DUMP(x)  cout << #x << " = " << (x) << endl;
#define FOR(i, m, n) for(ll i = m; i < n; i++)
#define IFOR(i, m, n) for(ll i = n - 1; i >= m; i-- )
#define REP(i, n) FOR(i,0,n)
#define IREP(i, n) IFOR(i,0,n)
#define FOREACH(x,a) for(auto& (x) : (a) )
#define ALL(v) (v).begin(), (v).end()
#define SZ(x) ll(x.size())

ll n;
vector<vector<vector<double>>> dp;

double rec(ll i, ll j, ll k){
  if(i==0 && j==0 && k==0) return 0;
  if(dp[i][j][k]>0) return dp[i][j][k];
  double a=0,b=0,c=0;
  if(i>0){
    a = rec(i-1,j+1,k);
    dp[i-1][j+1][k] = a;
  }
  if(j>0){
    b = rec(i,j-1,k+1);
    dp[i][j-1][k+1] = b;
  }
  if(k>0){
    c = rec(i,j,k-1);
    dp[i][j][k-1] = c;
  }
  return 1.0*(n+i*a+j*b+k*c)/(i+j+k);
}

int main(){
  cin >> n;
  vector<ll> cnt(4,0);
  REP(i,n){
    ll a; cin >> a;
    a = min(a,3LL);
    cnt[a]++;
  }
  // dp[i][j][k] = 0枚:i, 1枚:j, 2枚:k 種類からコンプまで必要なカード枚数の期待値
  dp = vector<vector<vector<double>>>(n+1, vector<vector<double>>(n+1, vector<double>(n+1,0.0)));
  cout << fixed << setprecision(10) << rec(cnt[0],cnt[1],cnt[2]) << endl;
}
0