二叉樹的子結構

帥地發表於2019-03-15
二叉樹

前言

牛客網劍指offer的66道題，刷起來！每道題會提供簡單的思路以及測試通過的程式碼

題目描述

輸入兩棵二叉樹A，B，判斷B是不是A的子結構。（ps：我們約定空樹不是任意一個樹的子結構）

二叉樹結構：

 class TreeNode {
     int val;
     TreeNode left;
     TreeNode right;
     TreeNode(int x) { val = x; }
 }

注：點選左下角的閱讀原文即可跳轉到原文，可以提交程式碼

解答思路

對於與二叉樹有關的題目，90% 是採取遞迴的方式來解決比較簡單的，這道題也是。

首先我們先以 A 的根節點 root1 作為起點來判斷 B 是否為 A的子結構。 如果是則直接返回 true，如果不是，則遞迴以 root1.left 和 root1.right 作為起點來判斷。程式碼如下：

public class 樹的子結構 {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        if (root2 == null || root1 == null) {
            return false;
        }
        // 判斷 B 是否為 A 的子結構
        return isSubTree(root1, root2);
    }

    // 判斷 B 是否為 A 的子結構
  private boolean isSubTree(TreeNode root1, TreeNode root2) {
      if (root1 == null) {
          return false;
      }// 以root1為root2的根節點，判斷子結構是否成立
      if (judge(root1, root2)) {
          return true;
      } else {
          // 如果root1作為起點不行，則遞迴判斷左右節點
          return isSubTree(root1.left, root2) || isSubTree(root1.right, root2);
      }
    }
    // 以root1為root2的根節點，判斷子結構是否成立
    private boolean judge(TreeNode root1, TreeNode root2) {
        if(root2 == null)
            return true;
        if(root1 == null)
            return false;
        if(root1.val == root2.val)
            return judge(root1.left, root2.left) && judge(root1.right, root2.right);

        return false;
    }
}

相關文章