80. 骰子的点数

最新推荐文章于 2025-08-24 11:06:10 发布

黄嘻嘻iiii

最新推荐文章于 2025-08-24 11:06:10 发布

阅读量248

点赞数

CC 4.0 BY-SA版权

分类专栏： OJ刷题总结文章标签：动态规划算法图论

本文链接：https://blog.csdn.net/m0_52394190/article/details/124872595

OJ刷题总结专栏收录该内容

17 篇文章

订阅专栏

在这里插入图片描述

思路

线性dp；
dp[i][j]表示i个骰子掷出数j的种数；
一个骰子可以掷出1~6，投掷i次得出数j可以表示为投掷i-1次得到j-k后，再投掷一次得到k（k取1至6），所以状态转移方程为
dp[i][j]+=dp[i-1][j-k], k=1~6;

代码

class Solution {
    
    
public:
    
    bool in(int i,int j)
    {
        return i>0&&j>=0;
    }

    vector<int> numberOfDice(int n) {
        int dp[100][100];
        for(int i=1;i<=6;i++)
            dp[1][i]=1;
        for(int i=2;i<=n;i++)
        {
            for(int j=i;j<=i*6;j++)
            {
                for(int k=1;k<7;k++)
                {
                    if(in(i-1,j-k))
                        dp[i][j]+=dp[i-1][j-k];
                }
                //cout<<dp[i][j]<<" ";
            }
            //cout<<endl;
        }
        vector<int> ans;
        for(int i=n;i<=6*n;i++)
            ans.push_back(dp[n][i]);
        return ans;
        
    }
};