n个骰子的点数(C++)

题目:

把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s的所有可能的值出现的概率。

你需要用一个浮点数数组返回答案,其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。

思路:

1. 递归方法:固定当前位,去向后递归,直到最后一颗骰子,然后向前归并每一种可能出现的结果,并使用一个哈希表存起来每种和值对应的次数

代码实现:

class Solution {
public:
    int max_item = 6;
    vector<double> twoSum(int n) {
        vector<double> result;
        if(n < 1)
            return result;
        int max_sum = n * max_item;
        int min_sum = n;
        int *sum = new int[max_sum - min_sum + 1]; // 使用哈希表保存每个和值出现的次数
        for(int i = min_sum; i <= max_sum; i++)
            sum[i - min_sum] = 0;
        Sum_item(n, sum);
        double total = pow(max_item, n);  // n个筛子的所有排列数
        for(int i = min_sum; i <= max_sum; i++)
            result.push_back((double)sum[i - min_sum] / total);
        return result;
    }
    void Sum_item(int n, int *sum)
    {
        for(int i = 1; i <= max_item; i ++)
            Sum_itemCore(n, n, i, sum);
    }
    void Sum_itemCore(int org, int cur, int cur_sum, int *sum)
    {
        if(cur == 1)
            sum[cur_sum - org]++;
        else
        {
            for(int i = 1; i <= max_item; i++)
                Sum_itemCore(org, cur - 1, cur_sum + i, sum); // 固定之前的筛子,向后递归
        }
    }
};

2. 使用循环方法,使用两个数组,和为n的数量为前面的n-1, n-2,n-3,n-4,n-5,n-6的和
代码实现:

class Solution {
public:
    int max_item = 6;
    vector<double> twoSum(int n) {
        vector<double> result;
        if(n < 1)
            return result;
        int *sum[2];
        sum[0] = new int [max_item * n + 1]; // 使用大数组保存所有可能的结果
        sum[1] = new int [max_item * n + 1];
        for(int i = 0; i < max_item * n + 1; i++)
        {
            sum[0][i] = 0;
            sum[1][i] = 0;
        }
        int flag = 0;
        for(int i = 1; i <= max_item; i++)
            sum[flag][i] = 1;
        for(int k = 2; k <= n; k++)
        {
            for(int i = 0; i < k; i++)
                sum[1 - flag][i] = 0;
            for(int i = k; i <= k * max_item; i++) // 和为k时有用的一段和值
            {
                sum[1 - flag][i] = 0;
                for(int j = 1; j <= i && j <= max_item; j++)
                    sum[1 - flag][i] += sum[flag][i - j]; // 和i大于6时和i为上一轮的n-1,n-2,n-3,n-4,n-5,n-6的和
            }
            flag = 1 - flag;
        }
        double total = pow(max_item, n);
        for(int i = n; i <= n * max_item; i++)
            result.push_back((double)sum[flag][i] / total);
        delete[] sum[0];
        delete[] sum[1];
        return result;
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值