1.问题描述
给一棵二叉树,找出从根节点到叶子节点的所有路径。
样例
给出下面这棵二叉树:
1
/ \
2 3
\
5
所有根到叶子的路径为:
[
"1->2->5",
"1->3"
]
2.解题思路
当一个节点的左右子树都为空时,把走到该节点的路径保存为向量,以此将所有叶子节点的路径保存在向量里。
3.代码实现
/**
* 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 the binary tree
* @return all root-to-leaf paths
*/
vector<string> binaryTreePaths(TreeNode* root) {
// Write your code here
vector<string>paths;
if(root==NULL)
return paths;
def(root,paths,to_string(root->val));
return paths;
}
void def(TreeNode*root,vector<string>&st,string strpaths)
{
if(root->left==NULL&&root->right==NULL){
st.push_back(strpaths);
return;
}
if(root->left!=NULL)
def(root->left,st,strpaths+"->"+to_string(root->left->val));
if(root->right!=NULL)
def(root->right,st,strpaths+"->"+to_string(root->right->val));
}
};
4.感想
因为需要一个大向量保存所有路径,所以需要重新定义一个函数,保存每一个路径,然后将这些路径保存在同一个向量里。