#130576 (C++14) No.108 トリプルカードコンプ

• 問題: No.108 トリプルカードコンプ
• ユーザー: はまやんはまやん
• 提出日時: 2016-11-15 03:50:51
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 7 ms / 5,000 ms
• コード長: 895 bytes
• コンパイル時間: 1,305 ms
• コンパイル使用メモリ: 167,044 KB
• 実行使用メモリ: 11,776 KB
• 最終ジャッジ日時: 2024-05-04 13:49:31
• 合計ジャッジ時間: 2,045 ms
• ジャッジサーバーID （参考情報）: judge4 / judge1

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	1 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	2 ms; 5,376 KB
testcase_05	AC	1 ms; 5,376 KB
testcase_06	AC	1 ms; 5,376 KB
testcase_07	AC	7 ms; 11,648 KB
testcase_08	AC	3 ms; 11,720 KB
testcase_09	AC	4 ms; 11,652 KB
testcase_10	AC	3 ms; 11,588 KB
testcase_11	AC	4 ms; 11,776 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	2 ms; 5,888 KB
testcase_14	AC	2 ms; 5,632 KB
testcase_15	AC	2 ms; 5,632 KB
testcase_16	AC	3 ms; 5,760 KB
testcase_17	AC	3 ms; 5,376 KB
testcase_18	AC	7 ms; 11,472 KB
testcase_19	AC	4 ms; 11,392 KB
testcase_20	AC	4 ms; 11,644 KB
testcase_21	AC	5 ms; 11,648 KB
testcase_22	AC	4 ms; 11,392 KB

ソースコード

#include<bits/stdc++.h>
using namespace std;
#define rep(i,a,b) for(int i=a;i<b;i++)



int N;
int A[101];
//-----------------------------------------------------------------
double memo[101][101][101];
double rec(int a, int b, int c) {
	if (0 <= memo[a][b][c]) return memo[a][b][c];
	int sm = a + b + c;
	if (sm == 0) return 0;

	double ret = ((double)N / (double)sm);
	
	if (0 < a) ret += rec(a - 1, b + 1, c) * ((double)a / (double)sm);
	if (0 < b) ret += rec(a, b - 1, c + 1) * ((double)b / (double)sm);
	if (0 < c) ret += rec(a, b, c - 1) * ((double)c / (double)sm);

	return memo[a][b][c] = ret;
}

int main() {
	cin >> N;
	rep(i, 0, N) cin >> A[i];

	int a = 0, b = 0, c = 0;
	rep(i, 0, N) {
		if (A[i] == 0)
			a++;
		else if (A[i] == 1)
			b++;
		else if (A[i] == 2)
			c++;
	}

	rep(i, 0, N + 1) rep(j, 0, N + 1) rep(k, 0, N + 1) memo[i][j][k] = -1;
	printf("%.10f\n", rec(a, b, c));
}