题目
给你二叉树的根节点 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]]
示例 2:
输入:root = [1,2,3], targetSum = 5
输出:[]
示例 3:
输入:root = [1,2], targetSum = 0
输出:[]
提示:
- 树中节点总数在范围
[0, 5000]
内 -1000 <= Node.val <= 1000
-
解题方法
递归 DFS
递归 DFS 遍历二叉树节点,记录路径以及当前路径对应的和。
时间复杂度O(n)
,空间复杂度O(n)
C++代码:/** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode() : val(0), left(nullptr), right(nullptr) {} * TreeNode(int x) : val(x), left(nullptr), right(nullptr) {} * TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {} * }; */ class Solution { public: void DFS(TreeNode* cur, int count, vector<int>& path, vector<vector<int>>& result) { if(!cur->left && !cur->right && count==0) { result.push_back(path); return; } if(!cur->left && !cur->right) return; if(cur->left) { path.push_back(cur->left->val); DFS(cur->left, count-cur->left->val, path, result); path.pop_back(); } if(cur->right) { path.push_back(cur->right->val); DFS(cur->right, count-cur->right->val, path, result); path.pop_back(); } return; } vector<vector<int>> pathSum(TreeNode* root, int targetSum) { vector<int> path; vector<vector<int>> result; if(!root) return result; path.push_back(root->val); DFS(root, targetSum-root->val, path, result); return result; } };