結果

問題 No.108 トリプルカードコンプ
ユーザー 0w1
提出日時 2017-01-06 17:33:53
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 11 ms / 5,000 ms
コード長 1,699 bytes
コンパイル時間 1,559 ms
コンパイル使用メモリ 167,728 KB
実行使用メモリ 12,636 KB
最終ジャッジ日時 2024-12-17 14:34:57
合計ジャッジ時間 2,401 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
typedef vector< int > vi;
typedef vector< vi > vvi;
typedef vector< ll > vl;
typedef vector< vl > vvl;
typedef pair< int, int > pii;
typedef vector< pii > vp;
typedef vector< double > vd;
typedef vector< vd > vvd;
typedef vector< string > vs;

template< class T1, class T2 >
int upmin( T1 &x, T2 v ){
  if( x > v ){
    x = v;
    return 1;
  }
  return 0;
}

template< class T1, class T2 >
int upmax( T1 &x, T2 v ){
  if( x < v ){
    x = v;
    return 1;
  }
  return 0;
}

const int INF = 0x3f3f3f3f;

const int MAXN = 100;

int N;
int A[ MAXN + 1 ];

void init(){
  cin >> N;
  for( int i = 0; i < N; ++i ){
    cin >> A[ i ];
  }
}

double dp[ MAXN + 1 ][ MAXN + 1 ][ MAXN + 1 ];
int vis[ MAXN + 1 ][ MAXN + 1 ][ MAXN + 1 ];

int cnt[ 3 ];

void preprocess(){
  for( int i = 0; i < N; ++i ){
    if( A[ i ] >= 3 ) continue;
    ++cnt[ A[ i ] ];
  }
}

double dfs( int c0, int c1, int c2 ){
  if( c0 == 0 and c1 == 0 and c2 == 0 ){
    return 0.0;
  }
  if( vis[ c0 ][ c1 ][ c2 ] ){
    return dp[ c0 ][ c1 ][ c2 ];
  }
  vis[ c0 ][ c1 ][ c2 ] = 1;
  double res = 1.0 * N / ( c0 + c1 + c2 );
  if( c0 - 1 >= 0 ){
    res += 1.0 * c0 / ( c0 + c1 + c2 ) * dfs( c0 - 1, c1 + 1, c2 );
  }
  if( c1 - 1 >= 0 ){
    res += 1.0 * c1 / ( c0 + c1 + c2 ) * dfs( c0, c1 - 1, c2 + 1 );
  }
  if( c2 - 1 >= 0 ){
    res += 1.0 * c2 / ( c0 + c1 + c2 ) * dfs( c0, c1, c2 - 1 );
  }
  return dp[ c0 ][ c1 ][ c2 ] = res; 
}

void solve(){
  double ans = dfs( cnt[ 0 ], cnt[ 1 ], cnt[ 2 ] );
  cout << fixed << setprecision( 10 ) << ans << endl;
}

signed main(){
  ios::sync_with_stdio( 0 );
  init();
  preprocess();
  solve();
  return 0;
}
0