- 路径总和
给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ \
7 2 1
返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。
class Solution {
public:
bool hasPathSum(TreeNode* root, int sum) {
if(root == NULL)
return false;
return test(root,0,sum);
}
bool test(TreeNode* root,int tempsum,int sum)
{
if(root == NULL)
return NULL;
if(root->left == NULL && root->right == NULL)
{
if(tempsum+root->val == sum)
return true;
else
return false;
}
return test(root->left,tempsum+root->val,sum)||test(root->right,tempsum+root->val,sum);
}
};
- 路径总和 II
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回
[
[5,4,11,2],
[5,8,4,5]
]
class Solution {
public:
vector<vector<int>> nodes;
vector<int> path;
vector<vector<int>> pathSum(TreeNode* root, int sum) {
if(root == NULL)
return nodes;
test(root,sum,0);
return nodes;
}
void test(TreeNode* root,int sum,int tempsum)
{
if(root == NULL)
return ;
path.push_back(root->val);
if(root -> left == NULL && root -> right == NULL)
{
if(tempsum + root -> val == sum)
nodes.push_back(path);
}
test(root -> left,sum,tempsum + root -> val);
test(root -> right,sum,tempsum + root -> val);
path.pop_back();
}
};
整个问题归根结底也就是树的前序遍历。