二项分布是数学的概念,定义及公式定理不在此说明.
注:递归版第三步及递推都依赖于C(M-1,N-1)+C(M-1,N)=C(M,N)的变形
递归版:
(我给递归版注释了1 2 3 是为了解析非递归版(递推版)中同等的操作
1 2 3)方便我们理解是怎么让递归变成递推的
public static double binomial(int N,int k,double p)
{
// 1
if(N == 0 && k == 0) return 1.0;
//2
if(N < 0 || k < 0) return 0.0;
//3
return (1.0 - p)*binomial(N - 1,k,p) + p * binomial(N - 1,k - 1,p);
}
递归版貌似很好理解,运行binomail(10,5,0.25)能通过,当我们用binomail(100,25,0.25),电脑似乎就不给力了,几天几夜恐怕也不会有结果,这时间复杂都有点捉急啊,那下面我们就用点小空间改成递推吧
非递归版(递推版)
public static double binomial(int N,int k,double p)
{
if (N < 0 || k < 0) return 0.0;
double[][] ret = new double[N + 1][k + 1];
//完成递归版第一步
ret[0][0] = 1.0;
//完成递归版第二步
for (int i = 1; i < N + 1; ++i)
{
ret[i][0] = ret[i - 1][0] * (1.0 - p);
}
for (int j = 1; j < k + 1; ++j)
{
ret[0][j] = 0.0;
}
//得到序列ret[N][k]
//完成递归版第三步
for (int i = 1; i < N + 1; ++i)
for (int j = 1; j < k + 1;++j)
{
ret[i][j] = (1.0 - p) * ret[i - 1][j] + p * ret[i - 1][j - 1];
}
return ret[N][k];
}
现在用非递归版运行binomial(100,50,0.25)是一秒的事了