LeetCode - 113 - 路徑總和 II

題目

• 難度：中等

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

• 語言： PHP

• 題目內容：

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

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

示例：

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

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

返回:

[
[5,4,11,2],
[5,8,4,5]
]

思路

深度優先遍歷（遞迴）,記錄遍歷路徑,如符合條件則輸出

<?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 {
    protected $res = [];
    /**
     * @param TreeNode $root
     * @param Integer $sum
     * @return Integer[][]
     */
    function pathSum($root, $sum) {
        $this->dfs($root, $sum, []);
        return $this->res;
    }

    /**
     * 深度優先遍歷
     */
    function dfs($root, $sum, $arr) {
        if ($root->val === null) {
            return [];
        }

        // 記錄遍歷路徑
        $arr[] = $root->val;
        if ($root->left === null && $root->right === null) {
            // 如條件符合,返回路徑
            if ($sum === $root->val) {
                $this->res[] = $arr;
                // return $this->res;
            }
            return [];
        }

        $sum -= $root->val;
        $this->dfs($root->left, $sum, $arr);
        $this->dfs($root->right, $sum, $arr);
    }
}

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

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