动态规划——丑数、n个骰子的点数

1.丑数

我们把只包含质因子 2、3 和 5 的数称作丑数（Ugly Number）。求按从小到大的顺序的第 n 个丑数。
题目链接

class Solution {
public:
    int nthUglyNumber(int n) {
        vector<int> dp(n,0);

        dp[0]=1;
        int two=0,there=0,five=0;

        for(int i=1;i<n;i++)
        {
            int num_two=dp[two]*2;
            int num_there=dp[there]*3;
            int num_five=dp[five]*5;
            dp[i]=fmin(fmin(num_two,num_there),num_five);//转移方程

        //当前下标乘(2,3,5)，已经用过当中的哪一个了值了
            if(dp[i]==num_two)
                two++;
            if(dp[i]==num_there)
                there++;
            if(dp[i]==num_five)
                five++;
        }
        return dp[n-1];
    }
};

2. n个骰子的点数

把n个骰子扔在地上，所有骰子朝上一面的点数之和为s。输入n，打印出s的所有可能的值出现的概率。
你需要用一个浮点数数组返回答案，其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。
题目链接

方法1：

class Solution {
public:
    vector<double> dicesProbability(int n) {
        //下一个骰子出现的所有点数=前面的一个筛子[+1+2+3+4+5+6];
        //每多一颗骰子，空间比前面需要多五个
        
        vector<vector<double>> dp;//建立动规数组
        dp.resize(n);//开辟n个二维空间

        for(int i=0;i<6;i++)//将只有一颗骰子时的所有概率初始化
        {
            dp[0].push_back(1.0/6.0);
        }

        for(int i=1;i<n;i++)//控制骰子数
        {
            dp[i].resize(dp[i-1].size()+5,0);//多五个空间，并且全部初始化为0

            for(int j=1;j<=6;j++)//控制点数
            {  
                int sub=j-1;//控制从第几个位置开始加起来
               
                 for(int k=0;k<dp[i-1].size();k++)//控制上一层所有的概率
                {
                    dp[i][sub++]+=dp[i-1][k]*1.0/6.0;
                }
            }
        }

        return dp[n-1];
    }
};

方法2：
由于dp[i]只与dp[i-1]有关系，因此我们只需要两个一维数组进行迭代即可

class Solution {
public:
    vector<double> dicesProbability(int n) {
        vector<double> ret(6,1.0/6.0);//当n为1的时候

        for(int i=1;i<n;i++)//控制骰子数
        {
            vector<double> temp(ret.size()+5,0); 
            for(int j=1;j<=6;j++)//控制骰子出现的点数
            {
                int sub=j-1;
                for(int k=0;k<ret.size();k++)//控制少一颗骰子时，所有数出现的概率
                {
                    temp[sub++]+=ret[k]/6;
                }
            }
            ret=temp;//进行迭代
        }
        return ret;
    }
};