题目描述
输入两颗二叉树A,B,判断B是不是A的子结构。
这实际上二叉树遍历算法的一种应用,要在原二叉树中查找是否具有某课子树,只需要判断每个节点是否都在二叉树中是否出现即可。所以需要先判断头结点,只有头结点符合要求才继续比较其子树是否符合,一样依次从头结点开始比较直到其左右子树进行比较,如果都符合则说明B是A的子结构。下面是基于这种思路实现的代码(已被牛客AC):
package com.rhwayfun.offer;
public class SubTree {
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
public boolean HasSubtree(TreeNode root1, TreeNode root2) {
boolean hasSubTree = false;
if(root1 != null && root2 != null){
if(root1.val == root2.val){
hasSubTree = nodesValEqual(root1,root2);
}
if(!hasSubTree){
hasSubTree = HasSubtree(root1.left, root2);
}
if(!hasSubTree){
hasSubTree = HasSubtree(root1.right, root2);
}
}
return hasSubTree;
}
private boolean nodesValEqual(TreeNode root1, TreeNode root2) {
if(root2 == null) return true;
if(root1 == null) return false;
if(root1.val != root2.val) return false;
return nodesValEqual(root1.left, root2.left) && nodesValEqual(root1.right, root2.right);
}
}