leetcode 257. 二叉树的所有路径(递归 + 回溯)

day17

思路:

前序遍历,因为要从根节点开始找。

回溯:

返还到调用函数之前的状态

先来看参数

path存储的是 单条路径,也就是从根结点到一个叶子结点的路径。

当到了叶子结点以后,就把路径插入到容器中。然后返回。遍历到最后一个叶子结点的时候,也就是递归结束的时候。

返回到上一层的时候,需要思考,有没有因为本层的递归而给上一层带来了变化,影响了上一层的结果。

在本题:

带来的变化只能有一个,res 中增加了一条路径。

所以当返回的时候,path要和之前调用的一样。这样才能从该结点继续向别的结点遍历。

 



 

void traversal(TreeNode* root,string path,vector<string>& res)

 这里的参数,path。用的是值传递。也就是上一层递归和下一层递归之间的path是没有联系的。所以当函数返回到上一层的时候,用的就是上一层之前的path。

这里的res 用的是引用传递。为的就是在一层层递归中,res的值不会“消失”。

而path是每条路径都不一样,所以用值传递。

再来看函数调用

if(root->left) traversal(root->left, path + "->", res);
if(root->right) traversal(root->right, path + "->", res);

这里就有回溯,因为这里的 " -> " 是直接在函数中的,所以如果返回到上一层的话,就直接会被清空。

如果 " -> " 不写在函数中,也就是直接写在外面

if (cur->left) {
    path += "->";
    traversal(cur->left, path, result); // 左
}
if (cur->right) {
    path += "->";
    traversal(cur->right, path, result); // 右
}

这样的话,每次返回到本层的时候,如果从左边到右边的时候, 会发现前面多加了 " -> " 。

因为改变了  path 的值

 

所以需要手动pop掉

if (cur->left) {
    path += "->";
    traversal(cur->left, path, result); // 左
    path.pop_back(); // 回溯 '>'
    path.pop_back(); // 回溯 '-'
}
if (cur->right) {
    path += "->";
    traversal(cur->right, path, result); // 右
    path.pop_back(); // 回溯 '>' 
    path.pop_back(); //  回溯 '-' 
}

 注意点:

1. 将 val 转为字符串

2. 将 “中” 放在前面。因为如果是叶子结点的话,需要先将叶子结点插入,再弹出。

3. 为了代码精简,可以将 "->" 放在函数的参数中,这样就不用手动的pop元素。

4.为了防止空结点操作,在调用之前都需要判断结点是否为空

class Solution {
private:
    void traversal(TreeNode* cur, string path, vector<string>& result) {
        path += to_string(cur->val); // 中,中为什么写在这里,因为最后一个节点也要加入到path中
        if (cur->left == NULL && cur->right == NULL) {  // 直接处理叶子
            result.push_back(path);
            return;
        }
        if (cur->left) {
            path += "->";
            traversal(cur->left, path, result); // 左

        }
        if (cur->right) {
            path += "->";
            traversal(cur->right, path, result); // 右
         
        }
    }

public:
    vector<string> binaryTreePaths(TreeNode* root) {
        vector<string> result;
        string path;
        if (root == NULL) return result;
        traversal(root, path, result);
        return result;

    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值