n个六面的骰子,扔一次之后和为k的概率是多少?

题目要求:n个六面的骰子,扔一次之后和为k的概率是多少?给定骰子数量n和骰子的点数之和k,求概率res。

思路:动态规划dp。

问题分析:每个骰子的点数都是1~6,因此n个骰子的总点数和的范围是n,6n,需要计算点数之和恰好等于k的方案数。

1.确定dp数组及其下标的含义:dpij表示使用i个骰子,总点数和为j的方案数。

2.确定递推公式:可知i个骰子的情况可以由i - 1个骰子的情况推出。

(1)因此dpij = dpi - 1j - 6 + dpi - 1\[j - 5 + dpi - 1j - 4 + dpi - 1j - 3 + dpi - 1j - 2 + dpi - 1j - 1。

(2)即:dpij = dpi - 1j - v for v = 1 to 6 。

3.dp数组如何初始化:dp00 = 1,表示使用0个骰子,总点数和为0的方案数为1。在这里没有实际意义,只用作dp数组的初始化。

4.确定遍历顺序:本题相当于分组背包问题。有n个组(每组对应一个骰子),每组有6个物品(点数1~6),每组必须且只能选1个物品,求恰好凑成总重量k的方案数。因此在使用二维数组时,组内物品的遍历顺序无所谓。

附代码:

java 复制代码
class Solution {
    public String probabilityOfSum(int n, int k) {
        // 如果 k 不在可能范围内,概率为 0
        if (k < n || k > 6 * n) {
            return "0";
        }

        // dp[i][s]:i 个骰子,点数和为 j 的方案数
        long[][] dp = new long[n + 1][6 * n + 1];
        dp[0][0] = 1;

        // i个骰子
        for (int i = 1; i <= n; i++) {
            // j表示i的骰子可能的点数和,范围为[i,6 * i]
            for (int j = i; j <= 6 * i; j++) {
                // v表示第i个骰子可选点数,范围为[1,6]
                for (int v = 1; v <= 6; v++) {
                    // j - v要大于等于0,因为第i个骰子的可选的点数一定不能超过点数和,否则需要排除
                    if (j - v >= 0) {
                        // 因为v会从1-6中取值,所以要把这6种可能累加
                        dp[i][j] += dp[i - 1][j - v];
                    }
                }
            }
        }

        // 总方案数 = 6^n
        long total = (long) Math.pow(6, n);
        // 可选方案数 = dp[n][k]
        long ways = dp[n][k];

        // 约分
        long g = gcd(ways, total);
        ways /= g;
        total /= g;

        return ways + "/" + total;
    }

    // 递归求最大公约数(欧几里得算法,即辗转相除法)
    private long gcd(long a, long b) {
        return b == 0 ? a : gcd(b, a % b);
    }
}

ACM模式:

java 复制代码
import java.util.Scanner;

class Solution {
    public String probabilityOfSum(int n, int k) {
        // 如果 k 不在可能范围内,概率为 0
        if (k < n || k > 6 * n) {
            return "0";
        }

        // dp[i][s]:i 个骰子,点数和为 j 的方案数
        long[][] dp = new long[n + 1][6 * n + 1];
        dp[0][0] = 1;

        // i个骰子
        for (int i = 1; i <= n; i++) {
            // j表示i的骰子可能的点数和,范围为[i,6 * i]
            for (int j = i; j <= 6 * i; j++) {
                // v表示第i个骰子可选点数,范围为[1,6]
                for (int v = 1; v <= 6; v++) {
                    // j - v要大于等于0,因为第i个骰子的可选的点数一定不能超过点数和,否则需要排除
                    if (j - v >= 0) {
                        // 因为v会从1-6中取值,所以要把这6种可能累加
                        dp[i][j] += dp[i - 1][j - v];
                    }
                }
            }
        }

        // 总方案数 = 6^n
        long total = (long) Math.pow(6, n);
        // 可选方案数 = dp[n][k]
        long ways = dp[n][k];

        // 约分
        long g = gcd(ways, total);
        ways /= g;
        total /= g;

        return ways + "/" + total;
    }

    // 递归求最大公约数(欧几里得算法,即辗转相除法)
    private long gcd(long a, long b) {
        return b == 0 ? a : gcd(b, a % b);
    }
}

public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        // 读取骰子个数 n 和目标和 k
        int n = scanner.nextInt();
        int k = scanner.nextInt();

        // 计算结果
        Solution solution = new Solution();
        String result = solution.probabilityOfSum(n, k);
        System.out.println(result);

        scanner.close();
    }
}
相关推荐
倒头就睡的小比特3 天前
算法竞赛C++常用的STL
c++·算法
小羊没烦恼!3 天前
初探性能优化——2个月到4小时的性能提升
java·开发语言·windows·算法·c#
猎头南楼3 天前
知识社区推荐系统实践:新用户冷启动与长短期兴趣建模的挑战 资深推荐算法工程师
人工智能·深度学习·算法·机器学习
旖旎夜光3 天前
力控面试题 01.01: 判定字符是否唯一(位运算) —— 题解
c++·学习·算法·leetcode·力控
wzdark3 天前
大规模并行计算中的负载均衡算法研究4
算法
Because_of_Her13 天前
并查集-听课笔记
笔记·算法·并查集
码流子3 天前
高速公路安全监测实践:碰撞监测预警+物联网底座,从感知到处置的闭环
大数据·人工智能·物联网·算法·架构
another heaven3 天前
【算法/C++ MD5算法能否逆解码?原理、C++实现与同类哈希算法对比】
c++·算法·哈希算法
wzdark3 天前
从算法设计模式看编程思维的抽象能力4
算法
2601_962218613 天前
万象生鲜系统称重自动多退少补算法解决生鲜非标品痛点
大数据·数据库·人工智能·python·算法