剑指 Offer 34. 二叉树中和为某一值的路径

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

叶子节点 是指没有子节点的节点。

 

示例 1:

输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]

一开始的做法:
class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int target) {
        vector<vector<int>> v;
        vector<int> vv;
        int sum = 0;
        getPath(root, target, sum, vv, v);
        return v;
    }
    void getPath(TreeNode* root, int target,int sum, vector<int> v1, vector<vector<int>> &v) {
        if (root == NULL)
            return;
        
        if (root->left == NULL && root->right == NULL) {
            if (sum + root->val == target) {
                v1.push_back(root->val);
                v.push_back(v1);
                return;
            }
            return;
        }
        v1.push_back(root->val);
        sum = sum + root->val;

        getPath(root->left, target, sum, v1, v);
        getPath(root->right, target, sum, v1, v);
    }
};

这个记过很慢,因为没有引用,后面看了题解,明白了如何用引用,通过回溯回弹,之前没有怎么刷过回溯的题目

class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int target) {
        vector<vector<int>> v;
        vector<int> vv;
        int sum = 0;
        getPath(root, target, sum, vv, v);
        return v;
    }
    void getPath(TreeNode* root, int target, int sum, vector<int> &v1, vector<vector<int>> &v) {
        if (root == NULL)
            return;

        v1.push_back(root->val);

        if (root->left == NULL && root->right == NULL) {
            if (sum + root->val == target) {
                v.push_back(v1);
            }
        }
        
        sum = sum + root->val;

        getPath(root->left, target, sum, v1, v);
        getPath(root->right, target, sum, v1, v);
        v1.pop_back();
    }
};

 

posted on 2022-02-21 16:58  4小旧  阅读(32)  评论(0)    收藏  举报

导航