递归
DFS用递归的形式,用到了栈结构,先进后出。
BFS选取状态用队列的形式,先进先出。
1.路径之和
给你二叉树的根节点 root 和一个表示目标和的整数 targetSum 。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum 。如果存在,返回 true ;否则,返回 false 。
输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22
输出:true
解题思路:递归(dfs)
假设从根节点到当前节点的和为val,可以转化成:是否从当前节点到到叶子节点的路径满足路径和为sum-val。
不难发现这满足递归的性质,若当前节点就是叶子节点,那么我们直接判断 sum 是否等于 val 即可(因为路径和已经确定,就是当前节点的值,我们只需要判断该路径和是否满足条件)。若当前节点不是叶子节点,我们只需要递归地询问它的子节点是否能满足条件即可。
struct TreeNode{
int val;
TreeNode* left;
TreeNode* right;
TreeNode(): val(),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:
bool hasPathSum(TreeNode* root, int sum) {
if(root==nullptr) return false;
if(root->left==nullptr && root->right==nullptr)
return sum==root->val;
return hasPathSum(root->left,sum-root->val)||hasPathSum(root->right,sum-root->val);
}
};
2.路径之和II
给出路径
输入: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> > res;
vector<int> path;
void dfs(TreeNode* root, int targetSum) {
if(root==nullptr) return;
path.push_back(root->val);
targetSum-=root->val;
if(root->left==nullptr&&root->right==nullptr&&targetSum==0)
res.push_back(path);
dfs(root->left,targetSum);
dfs(root->right,targetSum);
path.pop_back();
}
vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
dfs(root,targetSum);
return res;
}
};