给定一个二叉树,找出所有路径中各节点相加总和等于给定 目标值
的路径。
一个有效的路径,指的是从根节点到叶节点的路径。
样例
给定一个二叉树,和 目标值 = 5
:
1
/ \
2 4
/ \
2 3
返回:
[ [1, 2, 2], [1, 4]
]
解题思路:直接套用二叉树路径的代码进行一次特判是否加和是目标值即可。
class Solution { public: /** * @param root the root of binary tree * @param target an integer * @return all valid paths */ vector<vector<int>>Ans; vector<int>tem; void F(vector<int>x,int ans){ int sum =0; for(int i=0;i<x.size();i++)sum+=x[i]; if(sum==ans)Ans.push_back(x); } vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) { // Write your code here if(root==NULL)return Ans; tem.push_back(root->val); if(root->left!=NULL)binaryTreePathSum(root->left,target); if(root->right!=NULL)binaryTreePathSum(root->right,target); if(root->left==NULL&&root->right==NULL){ F(tem,target); } tem.erase(tem.begin()+tem.size()-1); return Ans; } };