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

题目描述

输入两棵二叉树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;
		}

	}
}