# LeetCode-面试题34-二叉树中和为某一值的路径
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例:
给定如下二叉树,以及目标和 sum = 22
,
5
/
4 8
/ /
11 13 4
/ /
7 2 5 1
返回
代码语言:javascript复制[
[5,4,11,2],
[5,8,4,5]
]
提示:
节点总数 <= 10000
# 解题思路
初始化2个全局数组,之后进行DFS递归遍历
对于TreeSearch函数:
- 递归参数:当前节点node,当前目标值tar
- 终止条件:若node为空,直接返回
- 递归工作: 路径更新:将node.val加入到路径path 目标值更新:tar = tar-node.val,当目标值为0时说明路径符合要求 路径记录:当node为叶子节点且路径值tar==0,则将path加入res 先序遍历:递归左右子节点 路径回溯:向上回溯前,需要将当前节点从路径path中删除,即删去path路径的最后一个节点值
# Java代码
代码语言:javascript复制/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
LinkedList<List<Integer>> res = new LinkedList<>();
LinkedList<Integer> path = new LinkedList<>();
public List<List<Integer>> pathSum(TreeNode root, int sum) {
TreeSearch(root,sum);
return res;
}
public void TreeSearch(TreeNode node,int tar){
if(node==null) return;
path.add(node.val);
tar-=node.val;
if(tar==0&&node.left==null&&node.right==null)
res.add(new LinkedList(path));
TreeSearch(node.left,tar);
TreeSearch(node.right,tar);
path.removeLast();
}
}
# Python代码
代码语言:javascript复制# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def pathSum(self, root: TreeNode, sum: int) -> List[List[int]]:
if not root: return [];
res,stack = [],[]
def TreeSearch(root,curSum):
stack.append(root.val)
curSum =root.val
if(curSum==sum) and not root.left and not root.right:
res.append(list(stack))
if root.left:
TreeSearch(root.left,curSum)
if root.right:
TreeSearch(root.right,curSum)
stack.pop()
TreeSearch(root,0)
return res