判断二叉树B是否为二叉树A的子结构

最新推荐文章于 2021-03-14 22:13:25 发布

原创最新推荐文章于 2021-03-14 22:13:25 发布 · 700 阅读

0 ·

CC 4.0 BY-SA版权

算法同时被 2 个专栏收录

15 篇文章

订阅专栏

数据结构

11 篇文章

订阅专栏

本文介绍了一种算法，用于判断一棵二叉树B是否为另一棵二叉树A的子结构。通过遍历A树的所有节点，并针对每个节点判断以该节点为根的子树是否与B树相等来实现。

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

题目描述

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

思路：遍历A树所有结点，分别判断以A树某个结点为根节点的子结构是否与B树结构相等

import java.util.Stack;

class TreeNode {
	int val = 0;
	TreeNode left = null;
	TreeNode right = null;

	public TreeNode(int val) {
		this.val = val;
	}

}

public class Solution {
	public boolean HasSubtree(TreeNode root1, TreeNode root2) {
		if (root2 == null)
			return false;
		Stack<TreeNode> s = new Stack<TreeNode>();
		boolean tag = false;
		if (root1 != null)
			s.push(root1);
		while (!s.empty()) {
			TreeNode node = s.pop();
			if (equal(node, root2)) {
				tag = true;
				break;
			}
			if (node.right != null)
				s.push(node.right);
			if (node.left != null)
				s.push(node.left);
		}
		return tag;
	}

	public boolean equal(TreeNode root1, TreeNode root2) {

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