拼多多笔试题：多多君住宿舍（概率DP）

最新推荐文章于 2024-08-12 11:20:08 发布

信仰..

最新推荐文章于 2024-08-12 11:20:08 发布

阅读量422

点赞数

分类专栏： DP 常见笔试编程题文章标签： 1024程序员节

本文链接：https://blog.csdn.net/haut_ykc/article/details/109259995

版权

DP 同时被 2 个专栏收录

104 篇文章 1 订阅

订阅专栏

常见笔试编程题

32 篇文章 11 订阅

订阅专栏

该博客通过概率动态规划（DP）解决了一个问题：在多多君有特定天使数和宿舍人数的情况下，计算某一天宿舍可能剩下的人数的概率分布。算法使用了三维动态规划数组，根据不同的天使数、剩余独睡天数和当前宿舍人数进行计算，并考虑了不同情况下的转移概率。最终求得总概率并输出结果。

摘要由CSDN通过智能技术生成

思路：概率dp，我们设dp[i][j][k]：表示前i天多多君有j天使一个人过夜并且第i天宿舍有k个人的概率。

public class Solution {

    static int n, m;
    static double a, b;

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        n = in.nextInt();
        m = in.nextInt();
        a = in.nextDouble();
        b = in.nextDouble();
        if (n == 1 || m == 1) {
            double x = 1.0;
            System.out.printf("%.8f\n", x);
        } else {
            double[][][] dp = new double[m + 1][m + 1][n + 1];
            dp[1][1][1] = 1.0;
            for (int d = 2; d <= m; d++) {
                for (int i = 1; i <= d; i++) {
                    for (int j = 1; j <= n; j++) {
                        for (int k = 1; k <= n; k++) {
                            if (j == 1)
                                dp[d][i][j] += dp[d - 1][i - 1][k] * work1(j, k);
                            else
                                dp[d][i][j] += dp[d - 1][i][k] * work2(j, k);
                        }
                    }
                }
            }
            double ans = 0;
            for (int i = 1; i <= m; i++)
                for (int j = 1; j <= n; j++)
                    ans += dp[m][i][j] * i;
            System.out.printf("%.8f\n", ans);
        }
    }

    private static double work1(int x, int y) {
        if (x == y)
            return 1.0 - a;
        else {
            if (y - x == 1)
                return n == 2 ? b : (1.0 - a) * b;
            else
                return b * b;
        }
    }

    private static double work2(int x, int y) {
        if (x == 2 && y == 1) return a;
        if (x == 2 && y == 2) return n == 2 ? (1.0 - b) : (1.0 - b) * (1.0 - a) + a * b;
        if (x == 2 && y == 3) return 2.0 * (1.0 - b) * b;
        if (x == 3 && y == 1) return 0;
        if (x == 3 && y == 2) return a * (1.0 - b);
        return (1.0 - b) * (1.0 - b);
    }
}