劍指offer 二叉樹中和為某一值的路徑(C++)
阿新 • • 發佈:2019-01-10
題目描述
輸入一顆二叉樹和一個整數,打印出二叉樹中結點值的和為輸入整數的所有路徑。路徑定義為從樹的根結點開始往下一直到葉結點所經過的結點形成一條路徑。#include<iostream> #include<vector> using namespace std; 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> >FindPath(TreeNode *root,int expectNumber) { if(root) dfsFind(root,expectNumber); return all; } vector<int > temp; vector<vector<int > > all; void dfsFind(TreeNode *node,int left) { temp.push_back(node->val); if(left-node->val==0 &&!node->left &&!node->right)//到達葉子結點,而且值剛好等於路徑之和 { all.push_back(temp); } else { if(node->left) dfsFind(node->left,left-node->val); if(node->right)dfsFind(node->right,left-node->val); } temp.pop_back();//如果不滿足路徑和為預期值 將最後一個值剔除 返回到上一層 //遞迴的時候 理不清怎麼辦?如下圖,搞清楚每個節點在進入容器的時候 他進行到哪一步 他的下一步是什麼就可以 } }; int main() { //構建二叉樹 /* 1 2 3 4 6 3 找出節點和為7的路徑 輸出為 1 2 4 1 3 3 */ Solution sol; vector<vector<int> >res; TreeNode *root=new TreeNode(1); root->left=new TreeNode(2); root->right=new TreeNode(3); root->left->left=new TreeNode(4); root->left->right=new TreeNode(6); root->right->left=new TreeNode(3); res=sol.FindPath(root,7); for(int i=0;i<res.size();i++) { for(int j=0;j<res[i].size();j++) { cout<<res[i][j]<<" "; } cout<<endl; } return 0; }