【LeetCode】113. Path Sum II

最新推荐文章于 2019-07-22 00:04:14 发布

原创最新推荐文章于 2019-07-22 00:04:14 发布 · 380 阅读

0 ·

CC 4.0 BY-SA版权

LeetCode 同时被 2 个专栏收录

85 篇文章

订阅专栏

DFS

35 篇文章

订阅专栏

本文探讨了在给定的二叉树中寻找所有从根到叶子节点的路径，其中路径上的元素和等于指定的数值。通过深度优先搜索(DFS)算法实现，详细解释了如何递归地遍历树并跟踪当前路径的总和。

Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.

Note: A leaf is a node with no children.

Example:

Given the below binary tree and sum = 22,

      5
     / \
    4   8
   /   / \
  11  13  4
 /  \    / \
7    2  5   1

Return:

[
   [5,4,11,2],
   [5,8,4,5]
]

题解：直接dfs即可，每次遇到叶节点时就判断一下是否和等于sum，如果则加入ans中，这里开始犯错是在root为空时才判断从而加了两次（因为左右都为空）

class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        vector<int> path;
        int tmp=0;
        dfs(root,tmp,path,sum);
        return ans;
    }
    void dfs(TreeNode* root,int tmp,vector<int>& path,int sum){
        if(root==NULL){
            return;
        }
        path.push_back(root->val);
        tmp+=root->val;
        if(!root->left&&!root->right&&tmp==sum)ans.push_back(path);
        dfs(root->left,tmp,path,sum);
        dfs(root->right,tmp,path,sum);
        path.pop_back();
    }
private:
    vector<vector<int>> ans;
    
};