給定一個二叉樹和一個目標和,找到所有從根節點到葉子節點路徑總和等於給定目標和的路徑。
說明: 葉子節點是指沒有子節點的節點。
示例:
給定如下二叉樹,以及目標和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
思路: 首先判斷終止條件是如果二叉樹的葉子結點等於剩餘要求的sum那麼這條路徑成立,傳入存儲的變量裏。用一個vector<int>存儲路徑,如果放進如的值不成立,則需要再拿出來。
C++代碼如下:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
void Sum(TreeNode* root, int sum,vector<int>& out, vector<vector<int>>& res)
{
if(root == NULL) return;
out.push_back(root->val);
if(root->val == sum && root->left ==NULL &&root->right == NULL)
{
res.push_back(out);
}
Sum(root->left,sum-root->val,out,res);
Sum(root->right,sum-root->val,out,res);
out.pop_back();
}
vector<vector<int>> pathSum(TreeNode* root, int sum) {
vector<vector<int>> res;
vector<int> out;
Sum(root,sum,out,res);
return res;
}
};