leetcode--257.Binary Tree Paths
阿新 • • 發佈:2017-08-04
eno ring 左右 深度優先 public 搜索 height back -1
使用深度優先搜索的方法遍歷每一條枝,遍歷完的終止條件為左右子樹都為空,這時候將這條枝的string 插入vector 中。
class Solution { public: // vector 一直插入string,所以為引用類型,string不能是引用類型 void dfs(TreeNode *root, vector<string> & result, string temp) { if(root->left==NULL&&root->right==NULL) result.push_back(temp); if(root->left) dfs(root->left,result,temp+"->"+to_string(root->left->val)); if(root->right) dfs(root->right,result,temp+"->"+to_string(root->right->val)); } vector<string> binaryTreePaths(TreeNode* root) { vector<string>result; if(root==NULL) return result; string temp=to_string(root->val); dfs(root, result, temp); return result; } };
第二種方法,在函數內部進行遞歸:
vector<string> binaryTreePaths(TreeNode *root) { vector<string> result; if(root==NULL) return result; if(root->left==NULL&& root->right==NULL) return {to_string(root->val)}; vector<string> left=binaryTreePaths(root->left); vector<string> right=binaryTreePaths(root->right); left.insert(left.end(),right.begin(),right.end()); vector<string>::iterator iter; for(iter=left.begin();iter!=left.end();iter++) { *iter=to_string(root->val)+"->"+*iter; } return left; }
leetcode--257.Binary Tree Paths