2017-07-03 153 views
1

我正在努力編寫這個遞歸程序,並想知道是否有人可以提供幫助。遞歸循環R

我想這個代碼迴歸方程:

for k=1,2,... 

beta(k)=k-sum_(i=0)^(k-1)Kchoosei*beta(i)*exp(-i(k-i)) 

我已經做到了在R上的手工方式,但希望把它放在一個函數。

beta0<-0 

beta1<-1-choose(1,0)*beta0*exp(-0*lambdaL*(1-0)) 
beta2<-2-choose(2,0)*beta0*exp(-0*lambdaL*(2-0))-choose(2,1)*beta1*exp(-1*lambdaL*(2-1)) 
beta3<-3-choose(3,0)*beta0*exp(-0*lambdaL*(3-0))-choose(3,1)*beta1*exp(-1*lambdaL*(3-1))-choose(3,2)*beta2*exp(-2*lambdaL*(3-2)) 
beta4<-4-choose(4,0)*beta0*exp(-0*lambdaL*(4-0))-choose(4,1)*beta1*exp(-1*lambdaL*(4-1))-choose(4,2)*beta2*exp(-2*lambdaL*(4-2))-choose(4,3)*beta3*exp(-3*lambdaL*(4-3)) 
+0

lambdaL沒有定義,而不是在你的一般方程 –

回答

1

您可以爲總和定義第二個循環。請注意,此處的索引從1開始,而不是0,這會導致「索引偏移」。

beta = numeric() 
beta[1] <- 0 

for (k in 1:10){ 
    beta[k+1] <- k 
    for (i in 0:(k-1)) 
    beta[k+1] <- beta[k+1] - choose(k, i)*beta[i+1]*exp(-i*(k-i)) 
} 

beta 
# [1] 0.000000 1.000000 1.264241 2.080705 3.247551 4.528104 5.748673 
# [8] 6.876234 7.941197 8.972749 9.987645 
0

我覺得你需要2個函數,因爲你需要測試版的每個以前的值作爲輸入,而你只需要一個輸出。這是我提出(與您lambaL調整,因爲在你的帖子有些含糊,它與LambdaL == 1作品):

beta_vec <- function(k){ 
    if(k == 0) 0 else { 
    beta_vec_old <- beta_vec(k-1) 
    c(beta_vec_old,sum(sapply(0:(k-1),function(i){1-choose(k,i)*beta_vec_old[i+1]*exp(-i*(k-i))}))) 
}} 

beta <- function(k){ 
    tail(beta_vec(k),1) 
} 

# > beta_vec(5) 
# [1] 0.000000 1.000000 1.264241 2.080705 3.247551 4.528104 
# > beta(5) 
# [1] 4.528104 

(編輯在代碼錯字)