Leetcode 113. Path Sum II

最新推荐文章于 2018-09-19 20:27:06 发布

vincentCheungM

最新推荐文章于 2018-09-19 20:27:06 发布

阅读量298

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/github_37606447/article/details/72819125

leetcode 专栏收录该内容

20 篇文章

订阅专栏

本文解析了LeetCode上的经典题目Path Sum II，介绍了一种使用深度优先搜索的方法来查找二叉树中所有从根节点到叶子节点的路径，这些路径的节点值之和等于给定的目标值。文章详细介绍了算法思路及其实现代码。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Leetcode 113. Path Sum II

source url

题目描述

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

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

输入：树，目标和
输出：从根到叶满足目标和的路径

思路

深度有限搜索，同时维护一个根到当前节点路径，并且在叶子节点处进行判断；检查是否符合目标和，如果符合则将路径添加到输出结果中。此处，如果能确定树中节点值均为正数的话，可以通过剪枝操作减少运算量。

代码

/**
 * 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<vector<int> > res;
    vector<vector<int> > pathSum(TreeNode* root, int sum) {
        if(root==NULL) return res;
        vector<int> emp;
        dfs(root,emp,sum);

        return res;
    }

    void dfs(TreeNode* root, vector<int> t, int sum){
        if(root->left==NULL&&root->right==NULL){
            if(root->val==sum){
                t.push_back(root->val);
                res.push_back(t);
                return;
            }
        }else{
            t.push_back(root->val);
            if(root->left!=NULL){
                dfs(root->left,t,sum-root->val);
            }
            if(root->right!=NULL){
                dfs(root->right,t,sum-root->val);
            }
        }
    }
};