LintCode 480. 二叉树的所有路径

原创于 2020-03-23 13:49:50 发布 · 153 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种用于查找二叉树中所有从根节点到叶子节点路径的算法。通过深度优先搜索(DFS)策略，该算法遍历整棵树并记录路径，最终返回所有可能的路径集合。

描述
中文
English

给一棵二叉树，找出从根节点到叶子节点的所有路径。

class Solution {
public:
    /**
     * @param root: the root of the binary tree
     * @return: all root-to-leaf paths
     */
vector<string>ans;

void dfs(TreeNode * node,string path)
{
    cout<<node->val<<endl;
    if(node==NULL) return ;
    if(path=="") path=path+to_string(node->val);
    else path =path+"->"+to_string(node->val);
    if(node->left!=NULL) dfs(node->left,path);
    if(node->right!=NULL) dfs(node->right,path);
    if(node ->left==NULL && node->right==NULL)  ans.push_back(path);
}

vector<string> binaryTreePaths(TreeNode * root)
{
    // write your code here
    if(root==NULL) return ans;
    dfs(root,"");
    return ans;
}

};