[Swift]LeetCode250.计数相同值子树的个数 $ Count Univalue Subtrees

★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★
➤微信公众号：山青咏芝（shanqingyongzhi）
➤博客园地址：山青咏芝（https://www.cnblogs.com/strengthen/）
➤GitHub地址：https://github.com/strengthen/LeetCode
➤原文地址：https://www.cnblogs.com/strengthen/p/10214859.html 
➤如果链接不是山青咏芝的博客园地址，则可能是爬取作者的文章。
➤原文已修改更新！强烈建议点击原文地址阅读！支持作者！支持原创！
★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★★

Given a binary tree, count the number of uni-value subtrees.

A Uni-value subtree means all nodes of the subtree have the same value.

For example:
Given binary tree,

              5
             / 
            1   5
           /    
          5   5   5

return 4.

---

给定一个二叉树，计算单值子树的数目。

单值子树意味着子树的所有节点都具有相同的值。

例如：

给定二叉树，

              5
             / 
            1   5
           /    
          5   5   5

返回4。

---

 public class TreeNode {
     public var val: Int
     public var left: TreeNode?
     public var right: TreeNode?
     public init(_ val: Int) {
         self.val = val
         self.left = nil
         self.right = nil
    }
 }

class Solution {
    func countUnivalSubtrees(_ root:TreeNode?) -> Int {
        var b:Bool = true
        return isUnival(root, &b)
    }
    
    func isUnival(_ root:TreeNode?,_ b:inout Bool) -> Int
    {
        if root == nil { return 0 }
        var l:Bool = true
        var r:Bool = true
        var res:Int = isUnival(root?.left, &l) + isUnival(root?.right, &r)
        b = l && r
        if root?.left != nil
        {
            b = b && (root!.val == root!.left!.val)
        }
        else
        {
            b = b && true
        }
        
        if root?.right != nil
        {
            b = b && (root!.val == root!.right!.val)
        }
        else
        {
            b = b && true
        }
        var num:Int = b ? 1 : 0
        return res + num
    }
}