这道题共有2种解法,递归法和迭代法
🎯一、递归法:
🟠(1)确定递归参数和返回类型
需要二叉树的根节点和sum返回值,所以需要两个参数,返回值类型为booleaan类型。
再来看返回值,递归函数什么时候需要返回值?什么时候不需要返回值?这里总结如下三点:
如果需要搜索整棵二叉树且不用处理递归返回值,递归函数就不要返回值。
如果需要搜索整棵二叉树且需要处理递归返回值,递归函数就需要返回值。
如果要搜索其中一条符合条件的路径,那么递归一定需要返回值,因为遇到符合条件的路径了就要及时返回。(本题的情况)
🟠(2)确定终止条件
不要去累加然后判断是否等于目标和,那么代码比较麻烦,可以用递减,让计数器count初始为目标和,然后每次减去遍历路径节点上的数值。如果最后count == 0,同时到了叶子节点的话,说明找到了目标和。如果遍历到了叶子节点,count不为0,就是没找到。
🟠(3)确定单层递归的逻辑
因为终止条件是判断叶子节点,所以递归的过程中就不要让空节点进入递归了。
递归函数是有返回值的,如果递归函数返回true,说明找到了合适的路径,应该立刻返回
class solution {
public boolean haspathsum(treenode root, int targetsum) {
if (root == null) {
return false;
}
targetsum -= root.val;
// 叶子结点
if (root.left == null && root.right == null) {
return targetsum == 0;
}
if (root.left != null) {
boolean left = haspathsum(root.left, targetsum);
if (left) { // 已经找到
return true;
}
}
if (root.right != null) {
boolean right = haspathsum(root.right, targetsum);
if (right) { // 已经找到
return true;
}
}
return false;
}
}
🎯二、迭代法:
🟠(1)首先想到使用栈去存储从根节点到叶子节点的值,当遍历到叶子节点时,读取栈顶元素相加到结果集并删除.当根节点为空的时候,返回false
;
🟠(2)否则,将根节点压入栈,当栈中不为空的时候,判断栈的大小,使用一个临时节点cur读取栈顶元素,让res=cur.val,再判断这个节点的左右孩子是否为空并且总和等于sum(为空则意味着到了叶子节点,遍历完了一条路径,那么她的总和等于sum吗),若为真则返回true,
🟠(3)否则,判断它有没有到达叶子节点,也就是说他还有左孩子或者右孩子,若左孩子不为空,则累计当前的节点值
,并将该节点压入到栈中,判断右子树同理。
class Solution {
public boolean hasPathSum(TreeNode root, int sum) {
Stack<TreeNode> s=new Stack<>();
if (root == null) {
return false;
}
s.push(root);
while(!s.isEmpty()){
int size=s.size();
for(int i=0;i<size;i++){
TreeNode cur=s.pop();
int res=cur.val;
if(cur.left==null&&cur.right==null&&res==sum){
return true;
}
if(cur.left!=null){
cur.left.val+=res;
s.push(cur.left);
}
if(cur.right!=null){
cur.right.val+=res;
s.push(cur.right);
}
}
}
return false;
}
}