劍指Offer-Java-樹的子結構

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);
    }
}