ifelse＆for循環R-幫助需要一個簡單的例子

Question

我對R很新，只是在玩它。我寫了一些代碼來計算雙方給出的可能的蒼鷺三角形的數量。正如我們許多人所知道的，蒼鷺三角形的所有邊和整個區域都是整數。在下面給出的代碼中，&b是整數。

heron <- function(a,b) 
{ 
    d = 0 
    for(c in abs(a-b)+1:a+b-1) 
    { 
    area <- ((1/4)*sqrt((a^2+b^2+c^2)^2-(a^4+b^4+c^4))) 
    d <- ifelse(area == floor(area), d+1, d) 
    } 

    return(d) 
} 

該代碼返回錯誤的值。例如，heron(5,5)=2但是當我插入a=5,b=5時，它給出的答案代碼爲0。查看代碼，很明顯，for循環和ifelse部分存在一些問題;但我無法弄清楚，所以這個問題。提前致謝。

Answer 1

你正在運行我假設的浮點數學錯誤。如果你走徹底的計算：

a <- 5 
b <- 5 

c我可以從你for loop代碼蒐集，但它不是做你認爲：

abs(a-b)+1:a+b-1 

至9 abs(a-b)+ (1 through a+b) - 1給出了數字5。而不是(abs(a-b) + 1) through (a + b - 1)

c <- 1:9 

area <- 1/4 * sqrt((a^2 + b^2 + c^2)^2 - (a^4 + b^4 + c^4)) 

> area 
[1] 9.185587 10.155048 11.592023 13.346348 15.309311 17.410485 19.605484 21.866070 24.173850 
> 

其中沒有一個是整數。因此，d停留在0.

我也放棄了for循環。您可以使用R的向量化特性來進行此計算。它重複a和b所需的次數以使area載體的長度與c的長度相同。

然後是這樣的：

length(area[area==floor(area)]) 

會給你鷺三角形的數量。其他更瞭解該公式的人必須告訴你其是否正確。

Answer 2

我在代碼中看到兩個錯誤。

第一：根據本Wiki Page公式是不正確的，你忘了2（在平方根）

二：在for循環中，你必須添加一些額外的支架（在讀10 + 1:12 != 11:12）

heron <- function(a,b) 
{ 
    d = 0 
    for(c in (abs(a-b)+1):(a+b-1)) 
    { 
    area <- ((1/4) * sqrt((a^2+b^2+c^2)^2 - 2 * (a^4+b^4+c^4))) 
    d <- ifelse(area == floor(area), d+1, d) 
    } 

    return(d) 
} 

heron(5, 5) 
[1] 2 
heron(13, 25) 
[1] 0 

沒有第二個解決方案循環

Heron <- function(a, b) { 
C <- (abs(a-b)+1):(a+b-1) 
Res <- sapply(C, function(i) 
        ((1/4) * sqrt((a^2+b^2+i^2)^2 - 2 * (a^4+b^4+i^4)))) 
length(Res[Res == floor(Res)]) 
} 
Heron(5, 5) 
[1] 2 
Heron(13, 25) 
[1] 0