問題描述:
給定一個二叉樹,找出所有路徑中各節點相加總和等於給定目標值
的路徑。
一個有效的路徑,指的是從根節點到葉節點的路徑。
樣例給定一個二叉樹,和 目標值 =
5
:
1
/ \
2 4
/ \
2 3
返回:
[
[1, 2, 2],
[1, 4]
]
解題思路:
建立一個二維向量,用以儲存符合要求的路徑,路徑表示爲一維向量的形式。首先判斷根節點的左右子樹是否爲空,將根節點的數值放入一維向量中,如果左右子樹都爲空判斷根節點的數值是否等於給定的數值,若等於給定的數值,則將其放入二維向量中。定義一個整數型t,用以計算二叉樹路徑的長度。如果左子樹不爲空,則將左子樹節點的數值加在t上,同時將左子樹節點放入一維向量中,然後繼續判斷,繼續調用函數,直到葉子節點,即左右子樹都爲空,然後判斷路徑長度是否等於給定的數值,若滿足條件,將這條路徑對應的一維向量放入二維向量中。對於右子樹的情況做相同操作。
代碼實現:
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param root the root of binary tree
* @param target an integer
* @return all valid paths
*/
vector<vector<int>>lz;
vector<vector<int>> binaryTreePathSum(TreeNode *root, int target) {
// Write your code here
if(root==NULL)
return lz;
vector<int>l;
func(root,root->val,target,l);
return lz;
}
void func(TreeNode *root,int t,int target,vector<int> l){
l.push_back(root->val);
if(root->left==NULL&&root->right==NULL){
if(t==target)
lz.push_back(l);
return;
}
int s;
if(root->left!=NULL){
s=t+root->left->val;
func(root->left,s,target,l);
}
if(root->right!=NULL){
s=t+root->right->val;
func(root->right,s,target,l);
}
}
};
解題感悟:
需要在給定的函數中調用一個新的函數,新函數寫在給定函數體外,若直接在給定的函數體內寫操作代碼,會出現restricted call錯誤。錯誤情況如下:
if(root==NULL) return lz;
else{
vector<int>l;int t=root->val;
l.push_back(root->val);
if(root->left==NULL&&root->right==NULL&&t==target){
lz.push_back(l);
return lz;
}
if(root->left!=NULL){
t+=root->left->val;
binaryTreePathSum(root->left,target);
}
if(root->right!=NULL){
t+=root->right->val;
binaryTreePathSum(root->right,target);
}
}
}
};