結果

問題 No.108 トリプルカードコンプ
コンテスト
ユーザー 37zigen
提出日時 2016-05-19 12:18:11
言語 Java
(openjdk 23)
結果
AC  
実行時間 189 ms / 5,000 ms
コード長 2,635 bytes
コンパイル時間 2,454 ms
コンパイル使用メモリ 79,104 KB
実行使用メモリ 52,924 KB
最終ジャッジ日時 2024-10-06 06:03:56
合計ジャッジ時間 6,819 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

package yukicoder;
import java.util.Scanner;
public class Main{
	public static void main(String[] args){
		new Main().solve();
	}
	//memo[i][j][k]:0枚のカードがi種類,1枚のカードがj種類,2枚のカードがk種類の状態から、完了まで引く回数の期待値
	double memo[][][];
	int N;
	void solve(){
		double time=System.currentTimeMillis();
		Scanner sc=new Scanner(System.in);
		N=sc.nextInt();
		int[] A=new int[N];

		int sum_0=0;//0枚のカードの種類数
		int sum_1=0;//1枚のカードの種類数
		int sum_2=0;//2枚のカードの種類数
		for(int i=0;i<N;i++){
			A[i]=sc.nextInt();
			if(A[i]==0)sum_0++;
			else if(A[i]==1)sum_1++;
			else if(A[i]==2)sum_2++;
		}
		memo=new double[N+1][N+1][N+1];
		for(int i=0;i<N+1;i++)for(int j=0;j<N+1;j++)for(int k=0;k<N+1;k++)
			memo[i][j][k]=-1;
		memo[0][0][0]=0;
		double ans=dp(sum_0,sum_1,sum_2);
		System.out.println(ans);
		//p[i][j][k]=0枚のカードがi種類,1枚のカードがj種類,2枚のカードがk種類の状態になる確率。
		//		double[][][] from=new double[N+1][N+1][N+1];
		//		double[][][] to=new double[N+1][N+1][N+1];
		//		from[sum_0][sum_1][sum_2]=1;
		//e:期待値
		//		double e=0;
		//		for(int t=1;t<=3000;t++){
		//			for(int i=0;i<N+1;i++){
		//				for(int j=0;j+i<N+1;j++){
		//					for(int k=0;k+i+j<N+1;k++){
		//						//						tr(from);
		//						//						System.out.println();
		//						to[i][j][k]+=from[i][j][k]*((double)(N-i-j-k)/(double)N);
		//						if(i>=1){
		//							to[i-1][j+1][k]+=from[i][j][k]*((double)i/(double)N);
		//						}
		//						if(j>=1){
		//							to[i][j-1][k+1]+=from[i][j][k]*((double)j/(double)N);
		//						}
		//						if(k>=1){
		//							if(i==0&&j==0&&k==1)
		//								e+=t*from[i][j][k]*((double)k/(double)N);
		//							to[i][j][k-1]+=from[i][j][k]*((double)k/(double)N);
		//						}
		//					}
		//				}
		//			}
		//			from=to;double[][][] d=new double[N+1][N+1][N+1];
		//			to=d;
		//		}
		System.err.println(System.currentTimeMillis()-time);
	}
	void tr(double[][][] p){
		int n=p.length;
		for(int i=0;i<n;i++){
			for(int j=0;j<n;j++){
				for(int k=0;k<n;k++){
					System.out.println(i+" "+j+" "+k+" "+p[i][j][k]);
				}
			}
		}
	}
	double dp(int i,int j,int k){
		if(memo[i][j][k]!=-1)return memo[i][j][k];
		int res=i+j+k;
		if(res==0)return 0;
		else{
			double e=((double)N)/((double)res);
			double ret=e;
			if(i>=1){
				ret+=dp(i-1,j+1,k)*((double)i/((double)res));
			}
			if(j>=1){
				ret+=dp(i,j-1,k+1)*((double)j/((double)res));
			}
			if(k>=1){
				ret+=dp(i,j,k-1)*((double)k/(double)res);
			}
			memo[i][j][k]=ret;
			return ret;
		}
	}
}
0