【剑指offer】刷题记录-树的子结构

最新推荐文章于 2024-03-22 22:43:55 发布

原创最新推荐文章于 2024-03-22 22:43:55 发布 · 221 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#剑指offer

博客提出判断二叉树B是否为A子结构的问题，约定空树不是任意树子结构。给出解题思路，分两步：先在树A中找与B根节点相同的节点R，用递归遍历二叉树；再判断以R为根的子树是否与B结构相同，同样用递归判断节点值是否相同。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

思路：我们可以分成两步，第一步在树A中找到和B的根结点一样的结点R，再判断树A中以R为根结点的子树是不是包含和树B一样的结构。这样，第一步实际上就是对二叉树的遍历，可以使用递归的思想。第二步也可以用递归的思想来考虑：如果结点R的值和子树B的根结点的值不相同，那么以R为根结点的子树和树B肯定没有相同的结构；如果它们的值相同，那么就递归的判断它们各自的左右结点的值是否相同。

public class Solution {
	class TreeNode {
		int val;
		TreeNode left = null;
		TreeNode right = null;

		TreeNode(int x) {
			val = x;
		}
	}

	public boolean HasSubtree(TreeNode root1, TreeNode root2) {
		boolean result = false;
		if (root2 == null) {
			return false;
		}
		if (root1 == null && root2 != null) {
			return false;
		}
		if (root1.val == root2.val) {
			result = doesTree1HasTree2(root1, root2);
		}
		if (!result) {
			result = HasSubtree(root1.left, root2);
			if (!result) {
				result = HasSubtree(root1.right, root2);
			}
		}
		return result;
	}

	public boolean doesTree1HasTree2(TreeNode root1, TreeNode root2) {
		if (root2 == null) {
			return true;
		}
		if (root1 == null && root2 != null) {
			return false;
		}
		if (root1.val == root2.val) {
			return doesTree1HasTree2(root1.left, root2.left) &&     
               doesTree1HasTree2(root1.right, root2.right);
		} else {
			return false;
		}

	}
}