leetcode113.路径总和II

给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

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

示例:

给定如下二叉树,以及目标和 sum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:

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

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/path-sum-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

  1. 定义一个二维向量res,如果存在头节点,将该头节点放入一维向量,然后将一维向量,与当前节点的左右子节点放入递归循环,如果该一维向量中所有加起来等于目标值,则将这个一维向量放入res中
  2. 定义一个函数pathSum(),传入参数root,sum,path,传进来的path为空的一维向量,先将root放入path中,如果sum等于path中所有元素之和将path放入res中,并且root无子节点。否则递归root->left,root->right
  3. 大致代码如下:
    	void pathSum(TreeNode* root, int sum, vector<int> path, int pathSum) {// 多定义了一个pathSum来记录path所有元素之和
        if(root != nullptr) {
            path.push_back(root->val);
            pathSum += root->val;
            if(sum == pathSum && root->right == nullptr && root->left == nullptr) {
                res.push_back(path);
            } else {
                pathSum(root->left, sum, path, pathSum);
                pathSum(root->right, sum, path, pathSum);
            }
        }
    }
    

最终测试通过代码

/**
 * 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) {
        vector<int> path;
        pathSum(root, sum, path, 0);
        return res;
    }
    void pathSum(TreeNode* root, int sum, vector<int> &path, int path_sum) { // pathSum改为path_sum
        if(root != nullptr) {
            path.push_back(root->val);
            path_sum += root->val;
            if(sum == path_sum && root->right == nullptr && root->left == nullptr) {
                res.push_back(path);
            } else {
                pathSum(root->left, sum, path, path_sum);
                pathSum(root->right, sum, path, path_sum);
            }
            path.pop_back(); // 每次都需要返回到父节点
        }
    }
};
发布了83 篇原创文章 · 获赞 12 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/qq_42317011/article/details/102531306