蓝桥杯 印章 Java实现

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

样例输入n和m：2 3

样例输出（保留四位小数：0.7500

数据规模和约定：1≤n，m≤20

解题思路：p = 1/n.

动规步骤：

dp定义：dp[i] [j] : i张印章凑齐j种印章数的概率。
递推公式：
i < j : 不可能凑齐，dp[i] [j] = 0

j = 1时（只需要凑齐一种即可，n种里面的随意一种：
i = 1时，dp[i] [j] = 1；
i > 1时，每个图案是dp[i] [j] = p ^ i; 有n种图案，则是p ^ i * n, 化简为p ^ (i - 1);

其他情况：
有重复的：前面的i- 1张就已经凑齐了j种，第i张就是重复前面的印章，这个时候前面已经出现过的j种印章再次出现的时候，就是j * p，这个时候就是dp[i] [j] = dp[i-1] [j] * (j/n) = dp[i-1] [j] *(j*p)

无重复的：前面的i- 1张就已经凑齐了j - 1种，第i张就是新凑齐的一种，只需要乘以以前没有出现的印章出现的概率，剩下没出现的印章数是n - (j - 1)种，概率就是(n - j + 1) * p，这个时候dp[i] [j] = dp[i-1] [j-1]*(n-(j-1))/n = dp[i-1] [j-1]*(n-j+1)*p

package LanQiao;

import java.io.BufferedInputStream;
import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(new BufferedInputStream(System.in));
        //输入n和m
        int n = in.nextInt();
        int m = in.nextInt();
        //定义p代表每一次的概率
        double p = 1.0 / n;
        //定义dp数组 dp[m][n] 买m张凑齐n种
        double[][] dp = new double[m + 1][n + 1];
        //i代表买了i张 j代表凑齐j种
        for (int i = 1; i <= m; i++) {
            for (int j = 1; j <= n; j++) {
                //因为i<j 不可能买了i张凑齐j种 所以dp=0；
                if (i < j) {
                    dp[i][j] = 0;
                } else if (j == 1) {
                    //如果j=1 买了i张凑齐j种 所以dp=p的i-1次方
                    dp[i][j] = Math.pow(p, i - 1);
                } else {
                    //其他情况 买了i张凑齐j种 第i张有两种情况 第一种和之前凑齐的一样 第二种和之前凑齐的不一样
                    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]);
    }
}