手続き型スタイルメソッドを機能スタイルに変換する

Question

以下のコードは、黒いツリーのルートから左下のノードを横切って黒ノードを数えたものです。黒のノードの数は可変blackに格納されます。

fun isBalanced1(): Boolean { 
    require(!isEmpty()) { "Cannot check empty tree for balance"} 
    var x = root 
    var black = 0 
    while(x != null) { 
     if(!isRed(x)) { 
      black++ 
     } 
     x = x.left 
    } 
    return isBalanced(root, black) 
} 

スタイルは手続きであり、それは大丈夫動作します。

今、もっと機能的なスタイルで同じことをどうやって変えることができますか？これは、ノードがnullではなく、その後、黒のものをフィルタリングし、すべての一致をカウントしながら、木を横断するシーケンスジェネレータを使用しています

fun isBalanced1(): Boolean { 
    require(!isEmpty()) { "Cannot check empty tree for balance" } 
    val black = 
      generateSequence(root) { node -> node.left } 
        .takeWhile { node -> node != null } 
        .filter { node -> !isRed(node) }.count() 
    return isBalanced(root, black) 
} 

：

これは私が思いついたものです。

これは、このツリートラバーサルコードを変換する正しい方法ですか、それともKotlinの方が良い方法ですか？

Answer 1

コメントで与えられたアドバイスに続いて、これはコードは次のようになります方法です。この方法count()

fun isBalanced(): Boolean { 
    require(!isEmpty()) { "Cannot check empty tree for balance" } 
    val blackCount = 
      generateSequence(root) { node -> node.left } 
        .count { !isRed(it) } 
    return isBalanced(root, blackCount) 
} 

は、読みやすくするために、独自の段落にあり、takeWhileは必要ありません。