【剑指 Offer 学习】【面试题 18： 树的子结构】【思路】

题目：

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

 

思路：

• 若根节点相等，利用递归比较他们的子树是否相等
• 若根节点不相等，则利用递归分别在左右子树中查找。

详见代码，未添加测试用例

 

代码：

package 树的子结构;

import javax.sound.midi.Soundbank;

/**
 * 输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）
 * 
 * @author Administrator
 *
 */
class TreeNode {
	int val = 0;
	TreeNode left = null;
	TreeNode right = null;

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

class Solution {

	public boolean HasSubtree(TreeNode root1, TreeNode root2) {

		boolean result = false;
		// 当 Tree1 和 Tree2 都不为零的时候，才进行比较。否则直接返回false
		if (root2 != null && root1 != null) {
			// 如果找到了对应 Tree2 的根节点的点
			if (root1.val == root2.val) {
				// 以这个根节点为为起点判断是否包含 Tree2
				result = isSubTree(root1, root2);
			}
			// 如果找不到，那么就再去 root1 的左儿子当作起点，去判断时候包含 Tree2
			if (!result) {
				result = HasSubtree(root1.left, root2);
			}

			// 如果还找不到，那么就再去 root1 的右儿子当作起点，去判断时候包含 Tree2
			if (!result) {
				result = HasSubtree(root1.right, root2);
			}
		}
		// 返回结果
		return result;
	}
	
	public boolean isSubTree(TreeNode root1, TreeNode root2) {

		// 如果Tree2已经遍历完了都能对应的上，返回true
		if (root2 == null) {
			return true;
		}
		// 如果Tree2还没有遍历完，Tree1却遍历完了。返回false
		if (root1 == null) {
			return false;
		}
		// 如果其中有一个点没有对应上，返回false
		if (root1.val != root2.val) {
			return false;
		}

		// 如果根节点对应的上，那么就分别去子节点里面匹配
		return isSubTree(root1.left, root2.left) && isSubTree(root1.right, root2.right);
	}
}