LeetCode-Path Sum II

LeetCode路径总和II题解

最新推荐文章于 2019-04-07 22:09:40 发布

原创最新推荐文章于 2019-04-07 22:09:40 发布 · 2.0k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#LeetCode

LeetCode OJ 专栏收录该内容

48 篇文章

订阅专栏

本文针对LeetCode上的路径总和II问题提供了一种有效的解决方案，采用递归方式遍历二叉树并记录所有从根节点到叶子节点且路径数值等于给定数值的路径。通过一个例子具体说明了如何寻找这样的路径，并提供了Java实现代码，分析了其时间复杂度为O(n)，空间复杂度为O(logn)。

题目：https://oj.leetcode.com/problems/path-sum-ii/

Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.

For example:
Given the below binary tree and sum = 22 ,

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

return

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

源码：Java版本
算法分析：时间复杂度O(n)，空间复杂度O(logn)。

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        List<List<Integer>> results=new ArrayList<List<Integer>>();
        Stack<Integer> path=new Stack<Integer>();
        pathSum(root,sum,path,results);
        return results;
    }
    
    private void pathSum(TreeNode root,int sum,Stack<Integer> path,List<List<Integer>> results) {
        if(root==null) {
            return;
        }
        path.push(root.val);
        if(root.left==null && root.right==null) {
            if(sum==root.val) {
                results.add((Stack<Integer>)(path.clone()));
            }
        }
        pathSum(root.left,sum-root.val,path,results);
        pathSum(root.right,sum-root.val,path,results);
        path.pop();
    }
}