024二叉树中和为某一值的路径(树)
题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
解题思路
深度遍历,递归运算
C++代码实现
C++ clang++3.9
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
vector<vector<int> > res;//所有路径
vector<int> trace;//暂存一条路径
if(root)
DFS(root,expectNumber,res,trace);
return res;
}
void DFS(TreeNode* root,int sum,vector<vector<int> >& res,vector<int>& trace){
trace.push_back(root->val);
if(!root->left&&!root->right)//判断叶子节点
if(root->val==sum)
res.push_back(trace);
if(root->left)
DFS(root->left,sum-root->val,res,trace);
if(root->right)
DFS(root->right,sum-root->val,res,trace);
trace.pop_back();//移除最后一个元素,深度遍历完一条路径后要回退,不然的话vector里面的元素只会增加不会减少。
}
};
运行时间:3ms
占用内存:480k
python代码实现
python 2.7.3
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# 返回二维列表,内部每个列表表示找到的路径
def __init__(self):
self.res=[]
def FindPath(self, root, expectNumber):
# write code here
if not root:
return self.res
self.DFS(root,expectNumber,[])
return self.res
def DFS(self,root,sum,temp):
if not root:
return
if (not root.left) and (not root.right) and (sum==root.val):
self.res.append(temp+[root.val])
if root.left:
self.DFS(root.left,sum-root.val,temp+[root.val])
if root.right:
self.DFS(root.right,sum-root.val,temp+[root.val])
运行时间:28ms
占用内存:5752k