[leetcode] 257. Binary Tree Paths

* Binary Tree Paths*

描述

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”]

我的代码

深搜，简单的递归：
和之前不一样的是，搜索到叶子结点就需要返回，同时要在叶子结点处将本次的路径保存进结果的vector。
之所以又写了一个dfs是为了可以在递归的过程中同时记录下路径。

/**
 * 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> rlt;
        dfs(root, "", rlt);
        return rlt;
    }

    void dfs(TreeNode* root, string spre, vector<string> &res)
    {
        if (!root) return;
        spre += to_string(root->val);
        if (!root->left && !root->right)
        {
            res.push_back(spre);
            return ;
        }
        spre += "->";
        if(root->left && root->right)
        {
            dfs(root->left,spre,res);
            dfs(root->right,spre,res);
        }
        else if(!root->left)
        {
            dfs(root->right,spre,res);
        }
        else if(!root->right)
        {
            dfs(root->left,spre,res);
        }
        return ;
    }
};