题解 | #牛牛的路径总和#

牛牛的路径总和

https://www.nowcoder.com/practice/45a878da071d43458b8b4cb2db8275b6

/**
 * struct TreeNode {
 *	int val;
 *	struct TreeNode *left;
 *	struct TreeNode *right;
 *	TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <algorithm>
#include <cstddef>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @param targetSum int整型 
     * @return int整型vector<vector<>>
     */
    void backtravel(vector<vector<int>>& res, vector<int>& path, TreeNode* root, int& targetSum)
    {
        if(root == nullptr){
            if (targetSum == 0) res.push_back(path);
            return;
        }
        path.push_back(root->val);
        targetSum -= root->val;
        if (root->left == nullptr) {
            backtravel(res, path, root->right, targetSum);
            if (root->right != nullptr) {
                path.pop_back();
                targetSum += root->right->val;
            }
        }

        else if (root->right== nullptr) {
            backtravel(res, path, root->left, targetSum);
            if (root->left != nullptr) {
                path.pop_back();
                targetSum += root->left->val;
            }
        }
        // 厚大
        else if(root->left->val < root->right->val){
            backtravel(res, path, root->left, targetSum);
            path.pop_back();
            targetSum += root->left->val;
            backtravel(res, path, root->right, targetSum);
            path.pop_back();
            targetSum += root->right->val;
        }
        else{
            backtravel(res, path, root->right, targetSum);
            path.pop_back();
            targetSum += root->right->val;
            backtravel(res, path, root->left, targetSum);
            path.pop_back();
            targetSum += root->left->val;
        }

        return;

    }

    vector<vector<int>> findPaths(TreeNode* root, int targetSum) {
        // write code here
        vector<vector<int>> res;
        vector<int> path;
        backtravel(res, path, root, targetSum);
        return  res;
    }
};

全部评论

相关推荐

Beeee0927:正确的建议
点赞 评论 收藏
分享
评论
1
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务