印章（动态规划问题dp）

package com.seal;

import java.util.Scanner;

//共有n种图案的印章，每种图案的出现概率相同。小A买了m张印章，求小A集齐n种印章的概率。

public class Main {
	/*
	 * 1.找到状态，由题目可知状态为两个,买了的印章个数i和集齐的印章的个数j;
	 * 2.明确dp数组的定义,dp[i][j]表示买了i个印章，集齐j个印章的概率;
	 * 当i<j时dp[i][j]=0;
	 * 
	 * j=1时
	 * 	i=1时 买1个印章,需要集齐1个印章的概率为1.
	 *  i>1时 买i个印章,需要集齐1个印章的概率为(1/n)^(i-1);
	 * 
	 * 3.根据选择,明确状态转移方程
	 * 我们买的第i张有两种选择，一是跟前面的i-1张有重复的,二是跟前面的i-1张没有重复的
	 * 
	 * ①有重复的，表明前面的i-1张已经凑齐了j种,那么这次是前面j种的概率为j/n; dp[i][j]=dp[i-1][j]*j/n;
	 * ②没有重复的，表明前面的i-1张已经凑齐了j-1种,那么这次不是前面j-1种的概率为 1-(j-1)/n; dp[i][j]=dp[i-1][j-1]*(1-(j-1)/n)
	 * 	 * 将两种情况相加就是我们的概率了。
	 * 
	 */
	public static void main(String[] args) {
		Scanner scan= new Scanner(System.in);
		int n=scan.nextInt();
		int m=scan.nextInt();
		
		if(m<n) {
			System.out.printf("%.4f",0.0);
			return;
		}
		if(m==0) {
			System.out.printf("%.4f",0.0);
			return;
		}
		
		double p=1.0/n;
		//dp数组
		double[][] dp=new double[m+1][n+1];
		
		dp[1][1]=1;
		for(int i=2;i<=m;i++) {
			dp[i][1]=Math.pow(p, i-1);
		}
		for (int i = 1; i <= m; i++) {
			for (int j = 2; j <=n; j++) {
				if(i<j) {
					dp[i][j]=0;
				}else {
					dp[i][j]=dp[i-1][j]*(j*p)+dp[i-1][j-1]*(1-(j-1)*p);
				}
			}
		}
		System.out.printf("%.4f",dp[m][n]);
		
	}
	
	
	
	
}