樹的子結構


輸入兩棵二叉樹 A,B,判斷 B 是否是 A 的子結構。咱們約定空樹不是任意一個樹的子結構java


基本思路:node

  • 首先在 A 中找到與 B 的根結點的值同樣的結點,能夠使用遞歸遍歷查找code

  • 找到的話,判斷 A 中該結點對應的子樹是否與 B 相等,相等返回 true,方法結束,不然返回 false,方法繼續執行遞歸

  • 繼續在 A 中遍歷查找與 B 的根結點的值一致的結點,找到的話重複第二步,若是不存在這樣的結點,直接返回 false同步


public class Solution {

    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        boolean result = false;
        // 當 Tree1 和 Tree2 都不爲 null 時,才進行比較。不然直接返回 false
        if(root2 != null && root1 != null) {
            // 若是找到了對應 Tree2 的根節點的點
            if(root1.val == root2.val) {
                // 以這個根節點爲起點判斷是否包含 Tree2
                result = doesTree1HaveTree2(root1, root2);
            }
            // 若是找不到,那麼就再去找 root 的左兒子當起點,判斷是否包含 Tree2
            if(!result) {
                result = HasSubtree(root1.left, root2);
            }
            // 若是還找不到,那麼就再去找 root 的右兒子當起點,去判斷是否包含 Tree2
            if(!result) {
                result = HasSubtree(root1.right, root2);
            }
        }
        // 返回結果
        return result;
    }

    public boolean doesTree1HaveTree2(TreeNode node1,TreeNode node2) {
        // 若是 Tree2 已經遍歷完了都能對應的上,返回 true
        // 注意這裏必須先判斷 node2,由於 Tree1 和 Tree2 是同步比較的,此時 node1 可能爲 null,也可能不爲 null
        if(node2 == null) {
            return true;
        }
        //若是 Tree2 尚未遍歷完,Tree1 卻遍歷完了,返回false
        if(node1 == null) {
            return false;
        }
        //若是其中有一個點沒有對應上,返回 false
        if (node1.val != node2.val) {
            return false;
        }
        // 若是根節點對應的上,那麼就分別去子節點裏面匹配
        return doesTree1HaveTree2(node1.left,node2.left) && doesTree1HaveTree2(node1.right,node2.right);
    }
}
相關文章
相關標籤/搜索