牛客網算法(十七):樹的子結構

題目描述

        輸入兩棵二叉樹A,B,判斷B是不是A的子結構。(ps:我們約定空樹不是任意一個樹的子結構)

題目分析

        第一種方法:判斷B樹是不是A樹的子結構,我們可以找出A中所有與B數根節點相等的元素,再以該元素爲根節點遍歷,判斷以該節點爲根節點的樹是否包含B樹的所有節點。
        第二種方法:可以創建兩個列表(ArrayList),分別儲存兩個樹的所有值(通過前序遍歷得到),然後使用ContainAll方法判斷列表A是否包含列表B。如果包含說明B樹是A樹的子結構。否則則反之。利用空間換時間。

Java實現

  1. 第一種方法:
	public static boolean HasSubtree(TreeNode root1, TreeNode root2) {
        boolean result = false;
        //當Tree1和Tree2都不爲零的時候,才進行比較。否則直接返回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 static boolean doesTree1HaveTree2(TreeNode node1, TreeNode node2) {
        //如果Tree2已經遍歷完了都能對應的上,返回true
        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);
    }
  1. 第二種方法:
import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> list1 = new ArrayList<Integer>();//用於儲存樹A的節點值
    public ArrayList<Integer> list2 = new ArrayList<Integer>();//用於儲存樹B的節點值
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
       //當Tree1和Tree2都不爲零的時候,才進行遍歷。否則直接返回false
        if(root1 == null || root2 == null){
            return false;
        }
        traverse1(root1);
        traverse2(root2);
        return list1.containsAll(list2);//判斷列表A是否包含列表B
    }
    //通過遞歸前序遍歷樹A
    public void traverse1(TreeNode root){
        if(root != null){
            list1.add(root.val);
            traverse1(root.left);
            traverse1(root.right);
        }
    }
    //通過遞歸前序遍歷樹B
    public void traverse2(TreeNode root){
        if(root != null){
            list2.add(root.val);
            traverse2(root.left);
            traverse2(root.right);
        }
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章