LeetCode - 112 - 路徑總和

題目

• 難度：簡單

• 題目連結：leetcode-cn.com/problems/path-sum/

• 語言： PHP

• 題目內容：

給定一個二叉樹和一個目標和，判斷該樹中是否存在根節點到葉子節點的路徑，這條路徑上所有節點值相加等於目標和。

說明: 葉子節點是指沒有子節點的節點。

示例：

給定如下二叉樹，以及目標和 sum = 22，

          5
         / \
        4   8
       /   / \
      11  13  4
     /  \      \
    7    2      1

返回true, 因為存在目標和為 22 的根節點到葉子節點的路徑5->4->11->2。

思路

深度優先遍歷（遞迴）

程式碼實現

<?php
/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     public $val = null;
 *     public $left = null;
 *     public $right = null;
 *     function __construct($value) { $this->val = $value; }
 * }
 */
class Solution {
    /**
     * @param TreeNode $root
     * @param Integer $sum
     * @return Boolean
     */
    function hasPathSum($root, $sum) {
        return $this->dfs($root, $sum);
    }

    /**
     * 深度優先遍歷
     */
    function dfs($root, $sum) {
        if ($root->val === null) {
            return false;
        }
        if ($root->left === null && $root->right === null) {
            return $sum === $root->val;
        }
        $sum -= $root->val;
        return $this->dfs($root->left, $sum) || $this->dfs($root->right, $sum);
    }
}

更多內容關注個人部落格：lemonlyue.github.io/

本作品採用《CC 協議》，轉載必須註明作者和本文連結