1.题目
2.思路
一开始看见这个,首先可以计算出最终数组的长度为n * 6 - n + 1;
然后思考一个问题,n最大为11, 每次都有6个数,如果直接暴力搜索,每次遍历6个数。遍历11次。就是Math.pow(6, 11) > 10^8,应该会超时,所以要想别的办法
然后想到了递归方法,第n次的值,其实是在第n - 1次的基础上,每个数都挨个加上1-6的概率,然后最终就得到了第n次的方法。这里的灵感来源于已知第一次的概率,求第二次的概率。两层for循环。
时间复杂度怎么计算?--O(n * n)
对第n次来说, 自身的循环有 6 * m , m 的大小等于(n - 1) * 6 - (n - 1) + 1,还要乘上循环n - 1轮
相当于O(n) = 6 * (5 * n - 4)* (n- 1) == O(n * n)
递推和递归
递归和递推往往都可以相互转化的。
递归:从后往前。先计算n, 需要计算n - 1, 然后最终计算1.能迭代递归,然后就是dfs
递推:从前往后。先计算1, 然后计算2,然后计算n.所以能写出递推公式,然后就是动态规划了。
class Solution {
public double[] dicesProbability(int n) {
// System.out.println(Math.pow(6, 11));
double a = 1.0 / 6;
double[]start = new double[]{a, a, a, a, a, a};
if(n == 1){
return start;
}
else{
double[]temp = dicesProbability(n - 1);
int m = temp.length;
double[]ans = new double[n * 6 - n + 1];
for(int i = 0 ; i < m ; i++){
for(int j = 0 ; j < 6; j++){
ans[i + j] += temp[i] * start[j];
// System.out.println(ans[i + j]);
}
}
return ans;
}
}
}