結果

問題 No.1688 Veterinarian
ユーザー Iván SotoIván Soto
提出日時 2021-09-24 22:17:20
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 182 ms / 3,000 ms
コード長 2,022 bytes
コンパイル時間 1,803 ms
コンパイル使用メモリ 207,692 KB
実行使用メモリ 66,340 KB
最終ジャッジ日時 2024-07-05 10:48:03
合計ジャッジ時間 2,890 ms
ジャッジサーバーID
(参考情報)
judge5 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 1 ms
6,940 KB
testcase_02 AC 2 ms
6,944 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 3 ms
6,940 KB
testcase_07 AC 62 ms
66,340 KB
testcase_08 AC 182 ms
66,336 KB
testcase_09 AC 164 ms
62,240 KB
testcase_10 AC 19 ms
13,728 KB
testcase_11 AC 25 ms
18,468 KB
testcase_12 AC 9 ms
13,732 KB
testcase_13 AC 4 ms
7,200 KB
testcase_14 AC 9 ms
11,680 KB
testcase_15 AC 11 ms
9,120 KB
testcase_16 AC 4 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

/**
 *  author: ivanzuki   
 *  created: Fri Sep 24 2021
**/
#include <bits/stdc++.h>

using namespace std;

const int S = 55;
const int M = 155;

double dp[S][S][S][S];

int main() {
  ios_base::sync_with_stdio(false);
  cin.tie(0);
  vector<vector<double>> Comb(M, vector<double>(M, 0));
  for (int i = 0; i < M; i++) {
    for (int j = 0; j <= i; j++) {
      if (j == 0 || i == j) {
        Comb[i][j] = 1;
      } else {
        Comb[i][j] = Comb[i - 1][j] + Comb[i - 1][j - 1];
      }
    }
  }
  int A, B, C, N;
  cin >> A >> B >> C >> N;
  dp[A][B][C][0] = 1;
  for (int a = A; a >= 0; a--) {
    for (int b = B; b >= 0; b--) {
      for (int c = C; c >= 0; c--) {
        for (int n = 0; n < N; n++) {
          // get 2 as
          if (a >= 2) {
            dp[a - 1][b][c][n + 1] += dp[a][b][c][n] * Comb[a][2] / Comb[a + b + c][2];
          }
          // get 2 bs
          if (b >= 2) {
            dp[a][b - 1][c][n + 1] += dp[a][b][c][n] * Comb[b][2] / Comb[a + b + c][2];
          }
          // get 2 cs
          if (c >= 2) {
            dp[a][b][c - 1][n + 1] += dp[a][b][c][n] * Comb[c][2] / Comb[a + b + c][2];
          }
          // get a and b
          if (a >= 1 && b >= 1) {
            dp[a][b][c][n + 1] += dp[a][b][c][n] * a * b / Comb[a + b + c][2];
          }
          // get b and c
          if (b >= 1 && c >= 1) {
            dp[a][b][c][n + 1] += dp[a][b][c][n] * b * c / Comb[a + b + c][2];
          }
          // get a and c
          if (a >= 1 && c >= 1) {
            dp[a][b][c][n + 1] += dp[a][b][c][n] * a * c / Comb[a + b + c][2];
          }
        }
      }
    }
  }
  double ans_a = 0, ans_b = 0, ans_c = 0;
  for (int a = 0; a <= A; a++) {
    for (int b = 0; b <= B; b++) {
      for (int c = 0; c <= C; c++) {
        ans_a += (A - a) * dp[a][b][c][N];
        ans_b += (B - b) * dp[a][b][c][N];
        ans_c += (C - c) * dp[a][b][c][N];
      }
    }
  }
  cout << setprecision(10) << fixed << ans_a << ' ' << ans_b << ' ' << ans_c << '\n';
  return 0;
}
0