swift 算法 简单25. 路径总和

径,这条路径上所有节点值相加等于目标和。

说明: 叶子节点是指没有子节点的节点。

示例: 
给定如下二叉树,以及目标和 sum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

解法:

/**
 * Definition for a binary tree node.
 * 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 hasPathSum(_ root: TreeNode?, _ sum: Int) -> Bool {
               guard  root != nil else{
            return false
        }
       
      
        if(root?.left == nil && root?.right == nil && root?.val == sum){
            return true;
        }
        if(root?.left == nil && root?.right == nil && root?.val != sum){
            return false;
        }
        return hasPathSum(root?.left,sum - (root?.val ?? 0)) || hasPathSum(root?.right,sum - (root?.val ?? 0));
        
    }

}

猜你喜欢

转载自blog.csdn.net/huanglinxiao/article/details/92560332
今日推荐