結果

問題 No.1688 Veterinarian
ユーザー Manuel1024Manuel1024
提出日時 2022-12-12 06:40:03
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 368 ms / 3,000 ms
コード長 1,461 bytes
コンパイル時間 1,055 ms
コンパイル使用メモリ 93,040 KB
実行使用メモリ 312,192 KB
最終ジャッジ日時 2024-04-24 00:01:15
合計ジャッジ時間 7,237 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 282 ms
312,192 KB
testcase_01 AC 278 ms
311,936 KB
testcase_02 AC 278 ms
312,192 KB
testcase_03 AC 281 ms
312,064 KB
testcase_04 AC 292 ms
312,064 KB
testcase_05 AC 284 ms
311,936 KB
testcase_06 AC 282 ms
312,064 KB
testcase_07 AC 279 ms
312,064 KB
testcase_08 AC 368 ms
311,936 KB
testcase_09 AC 360 ms
311,936 KB
testcase_10 AC 299 ms
312,192 KB
testcase_11 AC 289 ms
311,936 KB
testcase_12 AC 287 ms
311,936 KB
testcase_13 AC 285 ms
312,064 KB
testcase_14 AC 285 ms
312,064 KB
testcase_15 AC 306 ms
312,192 KB
testcase_16 AC 285 ms
312,192 KB
権限があれば一括ダウンロードができます

ソースコード

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