树的节点的和以及树的子树:
由于树的节点的求和运算,是一条路径判断求和,因此不太适合用容器思想。
子树是可以的,比较麻烦。采用前序遍历,先存储根节点判断;在存储左子节点判断,在存储右子节点判断,然后迭代替换。
//涉及到不是从根节点开始计算的题目;
应该从三部分计算:
1.根节点+左子节点+右子节点
2.重写根节点计算方法
求子树:
class Solution {
public boolean isSubtree(TreeNode s, TreeNode t) {
if(s==null) return false;
return isSubtreeWithRoot(s,t)||isSubtree(s.left,t)||isSubtree(s.right,t);
}
public boolean isSubtreeWithRoot(TreeNode s, TreeNode t){
if(s==null && t==null) return true;
if(s==null||t==null) return false;
if(s.val!=t.val)return false;
return isSubtreeWithRoot( s.left, t.left)&&isSubtreeWithRoot( s.right, t.right);
}
}
2.节点的和
一般都用sum-root.val
计算当前节点的值是否等于新节点的值
class Solution {
public int pathSum(TreeNode root, int sum) {
if(root==null) return 0;
int res=pathStartWithRoot(root,sum)+pathSum(root.left,sum)+pathSum(root.right,sum);
return res;
}
public int pathStartWithRoot(TreeNode root,int sum){
int res=0;
if(root==null) return 0;
if(root.val==sum)res++;
res+= pathStartWithRoot(root.left,sum-root.val)+pathStartWithRoot(root.right,sum-root.val);
return res;
}
}
3.判断路径和是否等于一个数
如果到达新的叶子节点;且值是我们要求的,则返回真
否则;返回新的求值
sum-root.val等
class Solution {
public boolean hasPathSum(TreeNode root, int sum) {
if(root==null)return false;
if(root.left==null&&root.right==null&&root.val==sum){
return true;
}
return hasPathSum(root.left,sum-root.val)||hasPathSum(root.right,sum-root.val);
}
}