粗看是一个全排列问题,递归排列的方式可以得出部分解,但是当n比较大时,消耗的时间会超出限制
题目要求并不关注全排列本身是怎么排列的列,只要求的得到解的数量。
可以想到当 n - 1 个骰子得到 m 个解后,加入第 n 个骰子,那么 n 个骰子对应的解就是 在 m 的基础上再加 6 种情况即可。
简单的来讲就是动态规划,从一个骰子的状态转移到 n 个骰子
转移方程为 dp[n][j + i] += dp[n - 1][j]; 其中n为骰子数,j 为上一个骰子的解,i 为 1-6 6种情况
代码由于是在递归排列失败后的基础上修改的,所以有个递归函数,其实并不需要,只要三重循环求 dp即可
class Solution {
int[] m;
public double[] dicesProbability(int n) {
double[] ans = new double[n*6 - n + 1];
int[] t = new int[n*6 + 1];
t[0] = 1;
find(t, n, 0);
int all = 0;
for(int i = n; i <= n*6; i++){
all += m[i];
}
for(int i = 0; i < n*6 - n + 1; i++){
ans[i] = m[n + i]*1.0 / all;
}
return ans;
}
void find(int[] t, int n, int deep){
if(deep == n){
m = t;
return;
}
int[] m = new int[n*6 + 1];
for(int i = 1; i <= 6; i++){
for(int j = 0;j <= n*6 - 6; j++){
m[j + i] += t[j];
}
}
find(m, n, deep + 1);
}
}