给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
[
[5,4,11,2],
[5,8,4,5]
]
code
/** * 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 { private: vector<vector<int>> res; private: void pathSumCore(TreeNode* root,int sum,vector<int>& tmp)//加&,节省内存,递归结束要回溯 { if(root==nullptr) { return ; } else if(root->left==nullptr&&root->right==nullptr) { if(root->val-sum==0) { tmp.push_back(root->val); res.push_back(tmp); tmp.pop_back();//回溯 } return ; } sum-=root->val; tmp.push_back(root->val); pathSumCore(root->left,sum,tmp); pathSumCore(root->right,sum,tmp); tmp.pop_back();//回溯 } public: vector<vector<int>> pathSum(TreeNode* root, int sum) { if(root==nullptr) return {}; vector<int> tmp; pathSumCore(root,sum,tmp); return res; } };
原文地址:https://www.cnblogs.com/tianzeng/p/12409164.html
时间: 2024-11-08 15:22:05