剑指offer(十七)——树的子结构

剑指offer(十七)——树的子结构

题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)

题解
首先理解题意,B是A的子结构的前提是,B树是A树的一部分,即从B的根节点开始,要与A其中一个结点开始,结点值完全一致。
在这里插入图片描述

  • 因为已经说了,空树不是子结构,所以先行判定两树是否为空。
  • 然后我们需要在A树中找到与B树根节点对应的节点,从那里开始进行判定,因此我们创建了一个函数IsSubtree(TreeNode,TreeNode)用来判定所有节点是否相符。
	public boolean HasSubtree(TreeNode root1,TreeNode root2) {
		if (root1 == null || root2 == null) {
			return false;
		}
		//从该节点开始判定是否相符
		//从A树的左孩子开始判定是否相符
		//从A树的右孩子开始判定是否相符
		return IsSubtree(root1, root2) || HasSubtree(root1.left, root2) || HasSubtree(root1.right, root2);
    }
	
	public boolean IsSubtree(TreeNode root1, TreeNode root2) {
	//当B树遍历完之后,即B树为A树子结构,返回true
		if (root2 == null) {
			return true;
		}
		//当A树遍历完,但是B树没有,证明不是子结构,返回false
		if (root1 == null) {
			return false;
		}
		//当该节点一致时,同时向其孩子搜索判定
		if (root1.val == root2.val) {
			return IsSubtree(root1.left, root2.left) && IsSubtree(root1.right, root2.right);
		}else {
			return false;
		}
	}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章