给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
返回:
[ [5,4,11,2], [5,8,4,5] ]
算法:dfs。这是标准的dfs题目。我们只需要看左子树或者右子树是否存在这样一条路径即可,若存在,则记录。
/** * 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<int>path; void dfs(TreeNode* root, int sum){ path.push_back(root->val); sum-=root->val; if(root->left||root->right){ if(root->left)dfs(root->left,sum); if(root->right)dfs(root->right, sum); } else{ if(!sum)res.push_back(path); } path.pop_back(); } vector<vector<int>> pathSum(TreeNode* root, int sum) { if(!root)return res; dfs(root,sum); return res; } };
原文:https://www.cnblogs.com/programyang/p/11167350.html