问题描述:给一棵二叉树,找出从根节点到叶子节点的所有路径。
样例
给出下面这棵二叉树:
1
/ \
2 3
\
5
所有根到叶子的路径为:
[
"1->2->5",
"1->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> path;
if(root==NULL) return path;
binaryTreePathCore(root,path,to_string(root->val));
return path;
}
void binaryTreePathCore(TreeNode *root,vector<string> &path,string strpath)
{if(root->left==NULL&&root->right==NULL)
{path.push_back(strpath);
return;
}
if(root->left!=NULL)
binaryTreePathCore(root->left,path,strpath+"->"+to_string(root->left->val));
if(root->right!=NULL)
binaryTreePathCore(root->right,path,strpath+"->"+to_string(root->right->val));
}
};
感想:开始时因为看的是课件上的程序,因为数据类型的问题一直出错,后来在网上找到这个比较容易懂。