結果

問題 No.1688 Veterinarian
コンテスト
ユーザー Manuel1024
提出日時 2022-12-12 06:40:03
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 361 ms / 3,000 ms
コード長 1,461 bytes
コンパイル時間 1,188 ms
コンパイル使用メモリ 93,532 KB
実行使用メモリ 312,064 KB
最終ジャッジ日時 2024-11-06 06:27:47
合計ジャッジ時間 7,081 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <cmath>
#include <iomanip>
using namespace std;

vector<vector<vector<vector<vector<double>>>>> mem;

vector<double> f(int n, int a, int b, int c){
    if(n == 0 || a+b+c <= 1) return {0.0, 0.0, 0.0};
    if(mem[0][n][a][b][c] >= 0.0){
        vector<double> res(3);
        for(int i = 0; i < 3; i++) res[i] = mem[i][n][a][b][c];
        return res;
    }
    for(int i = 0; i < 3; i++) mem[i][n][a][b][c] = 0.0;
    double z = (a+b+c)*(a+b+c-1);
    double y = z;
    y -= a*(a-1) + b*(b-1) + c*(c-1);
    vector<int> v = {a, b, c};
    for(int i = 0; i < 3; i++){
        if(v[i] < 2) continue;
        v[i]--;
        auto res = f(n-1, v[0], v[1], v[2]);
        res[i] += 1.0;
        for(int j = 0; j < 3; j++) mem[j][n][a][b][c] += res[j]*(v[i]+1)*v[i]/z;
        v[i]++;
    }
    auto res = f(n-1, a, b, c);
    for(int i = 0; i < 3; i++) mem[i][n][a][b][c] += res[i]*y/z;
    vector<double> ans(3);
    for(int i = 0; i < 3; i++) ans[i] = mem[i][n][a][b][c];
    return ans;
}

int main(){
    int a, b, c, n; cin >> a >> b >> c >> n;
    mem = vector<vector<vector<vector<vector<double>>>>>(3,
    vector<vector<vector<vector<double>>>>(55,
    vector<vector<vector<double>>>(55,
    vector<vector<double>>(55, vector<double>(55, -1)))));

    vector<double> ans = f(n, a, b, c);
    for(int i = 0; i < 3; i++){
        cout << fixed << setprecision(10) << ans[i] << " \n"[i == 2];
    }
    return 0;
}
0