科特林 - 轉換while循環功能的風格

我有以下的科特林功能：科特林 - 轉換while循環功能的風格

fun func(n: Int): Int { 

    var count = 1 

    var m = n 
    while(m != 1) { 

     m = if(m.isOdd()) 3 * m + 1 else m/2 

     count++ 
    } 

    return count 
}

我想編寫一個「功能性」的風格，簡單的算法，使用科特林的運營商如地圖（），計數（），等我能想出最接近的事是這樣的：

fun func(n: Int): Int { 

    return n.toList() 
      .map{ if(it.isOdd()) 3*it+1 else it/2 } 
      .takeWhile { it != 1 } 
      .count() 

}

顯然，上面的代碼不會因爲地圖只執行一次的工作，但你得到什麼，我想實現的想法。

PS：toList（）僅僅是一個擴展功能是一個int轉換爲包含INT列表：

fun Int.toList() = listOf(this)

來源

2017-08-17 Tiago

雖然@hotkey可能提供了正確的答案，但我認爲我們都應該同意，您的原始代碼實際上具有更高的內存效率，並且不會導致創建包含潛在數千個元素的序列，以查明生成了多少元素。 – Strelok

@Strelok這是一個很好的觀點。我在尋找的是一個解決方案，其中所有變量都是不可變的，而且熱鍵的答案能夠實現這一點。但是，如果有人知道如何做同樣的事情，但是「即時丟棄元素」（所以內存不是問題），請告訴我們。 – Tiago

既然你不知道有多少項目會出現，你可以構造一個（可能其中每個項目是基於前一個計算無限）序列，然後用你的病情it != 1限制，並指望有多少項目有：

return generateSequence(n) { if (it.isOdd()) 3 * it + 1 else it/2 } 
     .takeWhile { it != 1 } 
     .count()

這裏，generateSequence(n) { ... }，構建了一個Sequence<Int>有n爲它的第一個元素，下面的每個元素是通過作爲lambda傳遞的代碼來計算的（它是在前一個元素上調用的，並且只有在查詢另一個項目時，也就是懶惰地）。

來源

2017-08-17 00:22:11 hotkey

科特林 - 轉換while循環功能的風格

回答

相關問題