leetcode-树的子结构
基本思路
首先就是把每个节点都当做子结构的根节点来判断是否与第二个子树相等,第一个递归就是递归每一个节点,第二个递归则用来判断是否是子结构。
如果要判断是否是子结构,有三种情况。
1.如果AB都为null,则相等。
2.一个为null一个不为null,不相等。
3.都不为null,如果值不相等则不相等。如果值相等,则递归判断左右子树,此时如果B的一个子树为null,那么则认为该子树是相等的,因为只需要判断部分结构相符即可。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public boolean isSubStructure(TreeNode A, TreeNode B) {
return get(A,B);
}
public boolean get(TreeNode A, TreeNode B){
if(A == null || B == null) return false;
return get(A.left,B) || get(A.right,B) || getChild(A,B);
}
public boolean getChild(TreeNode A, TreeNode B){
if(A == null && B == null) return true;
else if(A != null && B != null){
if(A.val == B.val){
boolean leftBool = true;
boolean rightBool = true;
if(B.left != null){
leftBool = getChild(A.left,B.left);
}
if(B.right != null){
leftBool = getChild(A.right,B.right);
}
return true && leftBool && rightBool;
}else{
return false;
}
}else{
return false;
}
}
}
网上思路
基本相同,但是条件的判断更为简洁。
class Solution {
public boolean isSubStructure(TreeNode A, TreeNode B) {
return (A != null && B != null) && (recur(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B));
}
boolean recur(TreeNode A, TreeNode B) {
if(B == null) return true;
if(A == null || A.val != B.val) return false;
return recur(A.left, B.left) && recur(A.right, B.right);
}
}