【剑指offer】题43：n个骰子的点数

最新推荐文章于 2024-07-21 10:26:58 发布

xiaxzhou

最新推荐文章于 2024-07-21 10:26:58 发布

阅读量259

点赞数

分类专栏：剑指offer 动态规划

本文链接：https://blog.csdn.net/xiaxzhou/article/details/74396552

版权

剑指offer 同时被 2 个专栏收录

56 篇文章 0 订阅

订阅专栏

动态规划

24 篇文章 0 订阅

订阅专栏

动态规划
使用dp[k][i]表示k个骰子时，点数和为i的组合数
则转移状态方程：

d p [k] [i] = \sum j = 1 6 d p [k - 1] [i - j]

$dp[k][i]=\sum_{j=1}^6{dp[k-1][i-j]}$

void func(int n)
{
    if (n < 1)
    {
        return;
    }
    vector<int> vec(n*6+1);
    vector<vector<int>> dp(n+1,vec);

    for (auto i = 1; i <= 6;i++)
    {
        dp[1][i] = 1;
    }
    for (auto k = 1; k < n; ++k)
    {
        for (auto i = k; i < 6*k+1; ++i)
        {
            for (auto j = 1; j <= 6; j++)
            {
                dp[k + 1][i + j] += dp[k][i];
            }
        }
    }
    int sum = accumulate(dp[n].begin(), dp[n].end(),0);
    for (auto i = 0; i < 6 * n + 1;++i)
    {
        if (dp[n][i]!=0)
        {
            cout << i << " " << (double)dp[n][i] / (double)sum<<endl;
        }
    }
}