剑指Offer之二叉树的子结构

题目:

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

  B是A的子结构, 即 A中有出现和B相同的结构和节点值。

  例如:

给定的树 A:

     3
    / \
   4   5
  / \
 1   2
给定的树 B:

   4 
  /
 1
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值。

  示例 1:

输入:A = [1,2,3], B = [3,1] 
输出:false 

  示例 2:

输入:A = [3,4,5,1,2], B = [4,1] 
输出:true

 题解:

  先在A树中找到B的根节点,然后在一同左右遍历即可,直到B树为null。

public class TreeNode 
{
    public int val;
    public TreeNode left;
    public TreeNode right;
    public TreeNode(int x) { val = x; }
}
public class Solution 
{
    //先在A树中找到B的根节点
    public bool IsSubStructure(TreeNode A, TreeNode B) 
    {
        if( A == null || B == null)
        {
            return false;
        }
        if(A.val == B.val)
        {
            return isSub(A,B);
        }
        //递归查找
        return IsSubStructure(A.left,B) || IsSubStructure(A.right,B);
    }
    //一同遍历A树B的左右节点
    private bool isSub(TreeNode a,TreeNode b)
    {
        if(b == null)
        {
            return true;
        }
        if( a == null && b != null)
        {
            return false;
        }
        //当前A B树节点相同 遍历当前节点左右子节点
        if(a.val == b.val)
        {
            return isSub(a.left,b.left) && isSub(a.right,b.right);
        }
        return false;
    }
}

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