蓝桥杯—印章—DP

1、问题描述

问题描述

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

输入格式

　　一行两个正整数n和m

输出格式

　　一个实数P表示答案，保留4位小数。

样例输入

2 3

样例输出

0.7500

2、代码实现

import java.util.Scanner;

public class 印章 {

	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		int n = scanner.nextInt();
		int m = scanner.nextInt();
		
		//每个印章的概率
		double p = 1.0 / n;
		
		double[][] dp = new double[m + 1][n + 1];
		//购买的印章
		for(int i = 1;i <= m;i++) {
			//需要集齐的印章
			for(int j = 1;j <= n;j++) {
				//当我们购买的印章小于需要集齐的印章，那一定不能集齐
				if(i < j) {
					dp[i][j] = 0;
				}
				//当需要集齐的印章数量是1
				if(j == 1) {
					//当我们买一张时，那一定会集齐；当我们买多张时，概率就是 p ^ (i - 1)
					dp[i][j] = Math.pow(p, i - 1);
				}else {
					//第i张印章重复的概率+不重复的概率
					//当i - 1张印章集齐了j张印章，那么第i张印章重复的概率就是 j * p;如果j==m，那么重复的概率就是100%
					//当i - 1张印章集齐了j - 1张印章，那么第i张印章不重复的概率就是(n - (j - 1)) * p;需要集齐的印章-已经集齐的印章就是不重复的印章概率
					dp[i][j] = dp[i - 1][j] * j * p + dp[i - 1][j - 1] * (n - (j - 1)) * p;
				}
			}
		}
		
		System.out.printf("%.4f",dp[m][n]);
		
	}

}