結果

問題 No.108 トリプルカードコンプ
ユーザー junkaijunkai
提出日時 2020-11-26 00:27:41
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 12 ms / 5,000 ms
コード長 846 bytes
コンパイル時間 1,936 ms
コンパイル使用メモリ 169,644 KB
実行使用メモリ 14,208 KB
最終ジャッジ日時 2024-07-23 19:34:57
合計ジャッジ時間 3,059 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using P = pair<ll, ll>;
#define rep(i, n) for (int i = 0; i < (int)(n); i++)
double dp[110][110][110];
int main()
{
    int n;
    cin >> n;
    vector<int> v(4, 0);
    rep(i, n)
    {
        int a;
        cin >> a;
        if(a > 3) a = 3;
        v[a]++;
    }
    memset(dp, 0, sizeof(dp));
    function<double(int, int, int)> rec = [&](int i, int j, int k)
    {
        if(dp[i][j][k] > 0) return dp[i][j][k];
        if(k == n) return 0.0;
        double res = 0;
        if(i > 0) res += i * rec(i-1, j+1, k);
        if(j > 0) res += j * rec(i, j-1, k+1);
        if(n-i-j-k > 0) res += (n-i-j-k) * rec(i+1, j, k);
        res += n;
        res /= n-k;
        return dp[i][j][k] = res;
    };
    cout << fixed << setprecision(10) << rec(v[1], v[2], v[3]) << endl;
}
0