华为实习笔试2018.4.18（三）

三、有N个骰子，同时投掷出去，向上面的数字之和为A。

那么输入为N个筛子，请计算出A，和他出现的概率。

概率值，小数点保留5位。

输入描述：N，骰子数目

输出描述：

[[1， 0.16667], [2， 0.16667], [3， 0.16667], [4， 0.16667], [5， 0.16667], [6， 0.16667]]

（剑指offer的原题，但是输出格式变化，最后通过率60%）

#include <iostream>
#include <cmath>
#include <iomanip>
using namespace std;
const int FACE_NUM = 6;
int main()
{
	int number = 0;
	cin >> number;
	if (number <= 0)
		return -1;
	int size_dice = number * 6;
	double *probability_Sum = new double[size_dice + 1];
	double total = pow(6.0, number);
	int i, j, k;
	probability_Sum[0] = 0;
	for (i = 1; i <= FACE_NUM; i++)
		probability_Sum[i] = 1;
	for (; i <= size_dice; i++)
		probability_Sum[i] = 0;

	for (i = 2; i <= number; i++)
	{
		for (j = i * FACE_NUM; j >= i; j--)
		{
			probability_Sum[j] = 0;
			for (k = 1; k <= 6 && j >= k; k++)
			{
				probability_Sum[j] += probability_Sum[j - k];
			}
		}
		for (j = i - 1; j >= 0; j--)
			probability_Sum[j] = 0;
	}
	cout << "[";
	for (i = 0; i <= size_dice; i++)
	{
		if (probability_Sum[i] > 0.000009)
		{
			if (i < size_dice)
				cout << "[" << i << ", " << fixed << setprecision(5) << probability_Sum[i] / total << "]" << ", "<<" ";
			else
				cout << "[" << i << ", " << fixed << setprecision(5) << probability_Sum[i] / total << "]";
		}
	}
	cout << "]";
	return 0;
}