力扣刷题 DAY_42 二叉树

Leetcode257

链接:力扣

题目:

给定一个二叉树,返回所有从根节点到叶子节点的路径。

说明: 叶子节点是指没有子节点的节点。

示例1:

力扣刷题 DAY_42 二叉树_第1张图片

输入:root = [1,2,3,null,5]
输出2:["1->2->5","1->3"]

示例2:

输入:root = [1]
输出:["1"]

思路:

很经典的递归加回溯题目。我们同样可以在遍历的代码上进行修改解答此题,根据题目要求,我们要使用递归的前序遍历思路。

用一个vector数组result来记录最终答案,并用一个path数组记录当前已经走过的路径。

用前序遍历的方法。对于当前结点root,如果为空则return,若不为空,判断其是否为叶结点,如果非叶结点则将其加入path,依次递归进入其左子树和右子树,左右子树都递归完后还要从path里面删除该结点。

那么递归的边界在哪里呢?其实存在两个递归边界。

第一个边界就是当前结点为空时,直接return。

第二个边界就是当前访问结点为叶结点时,表示已经走完了一条路径,将该叶结点加入path,然后path里面的结点依次输出并转化为string数组,将其加入result来记录一条结果。然后该叶结点从path里面删除。

参考代码:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    void traversal(TreeNode *root,vector &path , vector &result) {
        if (root == nullptr) {
            return;
        }
        if (root->left == nullptr && root->right == nullptr) {
            path.push_back(root->val);
            string spath;
            for (int i = 0; i < path.size() - 1; i++) {
                spath += to_string(path[i]);
                spath += "->";
            }
            spath += to_string(path[path.size() - 1]);
            result.push_back(spath);
            path.pop_back();
        }
        else {
            path.push_back(root->val);
            traversal(root->left, path, result);
            traversal(root->right, path, result);
            path.pop_back();
        }
    }
    vector binaryTreePaths(TreeNode* root) {
        vector path;
        vector result;
        traversal(root, path, result);
        return result;
    }
};

你可能感兴趣的:(力扣刷题,leetcode,算法,二叉树)