目錄
問題重述:
題目:輸入一顆二叉樹的根節點和一個整數,打印出二叉樹節點值得和爲輸入整數得所有路徑。路徑定義爲從樹得根節點開始往下一直到葉節點所經過得節點形成一條路徑。
思路解析:
首先,注意一個雷區,題目要求一條路徑是從根節點到葉節點,因此,滿足條件得路徑一定會包含一個葉節點。
由於要求是從根到葉節點得路徑,因此可以想到利用二叉樹的先序遍歷,每往下遍歷一步,就記錄下路徑,同時在葉子節點處判斷是否滿足條件,當從低往上回溯的時候,應該將路徑序列中的最後一個元素捨棄。具體看下述代碼。
代碼實現:
/*
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> > buffer;
vector<int> tmp;
vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
if(root==NULL)
return buffer;
tmp.push_back(root->val);//添加路徑
//葉子節點判斷
if((expectNumber-root->val)==0 && root->left==NULL && root->right==NULL)
{
buffer.push_back(tmp);
}
FindPath(root->left,expectNumber-root->val);
FindPath(root->right,expectNumber-root->val);
//捨棄最後一個節點
if(tmp.size()!=0)
tmp.pop_back();
return buffer;
}
};