剑指Offer-Java-树的子结构

树的子结构


题目:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
代码:

package com.sjsq.test;

/**
 * @author shuijianshiqing
 * @date 2020/5/19 20:47
 */

/**
 * 输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
 */

public class Solution {

    public boolean HasSubtree(TreeNode root1,TreeNode root2){
        boolean result = false;
        if(root1 != null && root2 != null){
            if(root1.val == root2.val){
                result = HasTree(root1,root2);
            }
            if(!result){
                result = HasTree(root1.left,root2);
            }
            if(!result){
                result = HasTree(root1.right,root2);
            }
        }
        return result;
    }

    public boolean HasTree(TreeNode root1,TreeNode root2){

        //首先是递归结束条件,递归到B没有子节点时,说明比较完了,B是A子结构
        //遍历到A没有子节点时,说明还有B的子孙节点没有比较,此时B不是A的结构
        //这两句的顺序不能颠倒,因为先判断的B树是否还有节点。
        if(root2 == null){
            return true;
        }
        if(root1 == null){
            return false;
        }
        if(root1.val != root2.val){
            return false;
        }

        //到这里说明传入的两颗树的根节点值相同
        return HasTree(root1.left,root2.left) && HasTree(root1.right,root2.right);
    }
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章