【二叉树算法题记录】二叉树的所有路径,路径总和——回溯

目录

257. 二叉树的所有路径

题目描述

给你一个二叉树的根节点root ,按任意顺序,返回所有从根节点到叶子节点的路径。

题目分析

其实从根节点往下走,遍历的思路就如下图所示。我们先走到叶子节点(这是一条路径),然后再往上回溯 ,如果回溯到的上层节点有右孩子,那么就再按照右边的路径往下走(另一条路径)。

  1. 递归传入参数及返回值
cpp 复制代码
void traversal(TreeNode* cur, vector<int>& path, vector<string>& result)

cur是当前节点,path用来记录当前路径,result是最终返回的结果。

  1. 递归终止条件
    当我们遍历到叶子节点时,就要把当前路径path加入result了。
cpp 复制代码
if(cur->left == NULL && cur->right == NULL) {
            // 如果当前节点是叶子节点,也就是左右孩子都为空
            // 也就是说已经遍历了一条根节点到叶子节点的路径
            // 我们要把这条路径放进result中
            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);
            return;
        }
  1. 单层递归逻辑
    我们进行的是一个前序遍历,但是需要注意的是当前节点的处理要在递归终止条件之前进行:
cpp 复制代码
path.push_back(cur->val);   // 先将当前节点放进当前path中,再判断结束条件

这是因为所有节点包括叶子节点都要加入path,如果先判断终止,再处理当前节点则会漏掉叶子节点。

在进行左右孩子的遍历时,我们要注意判断节点是否为空:

cpp 复制代码
if(cur->left) {
            traversal(cur->left, path, result);
            path.pop_back();    // 回溯
        }
        if(cur->right) {
            traversal(cur->right, path, result);
            path.pop_back();    // 回溯
        }

cpp代码

cpp 复制代码
/**
 * 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* cur, vector<int>& path, vector<string>& result){
        path.push_back(cur->val);   // 先将当前节点放进当前path中,再判断结束条件
        if(cur->left == NULL && cur->right == NULL) {
            // 如果当前节点是叶子节点,也就是左右孩子都为空
            // 也就是说已经遍历了一条根节点到叶子节点的路径
            // 我们要把这条路径放进result中
            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);
            return;
        }
        
        if(cur->left) {
            traversal(cur->left, path, result);
            path.pop_back();    // 回溯
        }
        if(cur->right) {
            traversal(cur->right, path, result);
            path.pop_back();    // 回溯
        }
        
    }

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

112. 路径总和

题目描述

给你二叉树的根节点root和一个表示目标和的整数targetSum。判断该树中是否存在根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和targetSum。如果存在,返回true;否则,返回false

题目分析

其实这题跟上题是一样的,只不过这里多了一个targetSum的判断。

cpp代码

cpp 复制代码
/**
 * 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* cur, vector<int>& path, vector<int>& sum){
        path.push_back(cur->val);

        if(cur->left == NULL && cur->right == NULL) {

            cout << "当前路径: ";
            for(int p : path) {
                cout << p << ' ';
            }
            cout << endl;

            sum.push_back(accumulate(path.begin(), path.end(), 0));
            cout << "当前路径总和: " << accumulate(path.begin(), path.end(), 0) << endl;
            return;
        }

        if(cur->left) {
            traversal(cur->left, path, sum);
            path.pop_back();
        }
        if(cur->right) {
            traversal(cur->right, path, sum);
            path.pop_back();
        }
    }

    bool hasPathSum(TreeNode* root, int targetSum) {
        vector<int> path;
        vector<int> sum;
        if(root == NULL) return false;
        traversal(root, path, sum);
        if(find(sum.begin(), sum.end(), targetSum) == sum.end()){
            return false;
        }
        else return true;
    }
};
相关推荐
乐迪信息1 小时前
乐迪信息:基于AI算法的煤矿作业人员安全规范智能监测与预警系统
大数据·人工智能·算法·安全·视觉检测·推荐算法
hsjkdhs2 小时前
C++之多层继承、多源继承、菱形继承
开发语言·c++·算法
立志成为大牛的小牛2 小时前
数据结构——十七、线索二叉树找前驱与后继(王道408)
数据结构·笔记·学习·程序人生·考研·算法
星空下的曙光2 小时前
Node.js crypto模块所有 API 详解 + 常用 API + 使用场景
算法·node.js·哈希算法
Algo-hx2 小时前
数据结构入门 (七):从“链接”到“分支” —— 初探树与二叉树
数据结构
小贾要学习3 小时前
【数据结构】C++实现红黑树
数据结构·c++
StarPrayers.4 小时前
旅行商问题(TSP)(2)(heuristics.py)(TSP 的两种贪心启发式算法实现)
前端·人工智能·python·算法·pycharm·启发式算法
爱吃橘的橘猫4 小时前
嵌入式系统与嵌入式 C 语言(2)
c语言·算法·嵌入式
235164 小时前
【LeetCode】146. LRU 缓存
java·后端·算法·leetcode·链表·缓存·职场和发展
weixin_307779135 小时前
使用Python高效读取ZIP压缩文件中的UTF-8 JSON数据到Pandas和PySpark DataFrame
开发语言·python·算法·自动化·json