问题描述
Given a binary tree, return all root-to-leaf paths.
For example, given the following binary tree:
1
/ \
2 3
\
5
All root-to-leaf paths are:
[“1->2->5”, “1->3”]
思路分析
给一二叉树,返回从根结点到每一个叶子结点到路径。
递归整个二叉树,直到叶子结点(叶子结点没有子树),将结果push进vector,最后返回即可。
代码
递归
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<string> binaryTreePaths(TreeNode* root) {
vector<string> result;
if (root == NULL)
return result;
searchBT(root, to_string(root->val), result);
return result;
}
void searchBT(TreeNode* root, string path, vector<string>& result) {
if(root->left == NULL && root->right == NULL){
result.push_back(path);
return;
}
if(root->left != NULL)
searchBT(root->left, path + "->" + to_string(root->left->val), result);
if(root->right != NULL)
searchBT(root->right, path + "->" + to_string(root->right->val), result);
}
};
时间复杂度:
O(logn)
空间复杂度:
O(logn)
反思
重大失误,没有在函数中加&引用符号,导致每次递归都是在一个新的vector上操作,结果返回值为空。基础不牢啊!