剑指offer系列17之树的子结构

要求

• 时间限制：1秒
• 空间限制：32768K
• 热度指数：364970

题目描述

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

思路

HasSubtree()函数用来判断一个大的树中是否包含所给的子结构；而isSubtree()函数用来判断所给子结构是否是所给的树的子结构并且二者根结点是一样的。

python实现

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def HasSubtree(self, pRoot1, pRoot2):
        # write code here
        if pRoot1 is None or pRoot2 is None:
            return False
        return self.isSubtree(pRoot1, pRoot2) or self.HasSubtree(pRoot1.left, pRoot2) or self.HasSubtree(pRoot1.right, pRoot2)
    
    def isSubtree(self, pRoot1, pRoot2):
        # 树2刚进来时不为空，循环几次后为空还没遇到中止条件说明满足条件，此时1是否为空都无所谓
        if pRoot2 is None:
            return True
        # 树1先于树2被遍历完，一定不满足条件
        if pRoot1 is None:
            return False
        # 根节点相同并且左子树和左子树都满足这个条件，一直递归
        return pRoot1.val == pRoot2.val and self.isSubtree(pRoot1.left, pRoot2.left) and self.isSubtree(pRoot1.right, pRoot2.right)

• 运行时间：25ms
• 占用内存：5756k

C++实现

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if (pRoot1 == NULL || pRoot2 == NULL)
            return false;
        return isSubTree(pRoot1, pRoot2) || HasSubtree(pRoot1->left, pRoot2) 
            || HasSubtree(pRoot1->right, pRoot2);
    }
    // 判断2是否是1的子结构（从根节点处的比较）
    bool isSubTree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        // 这里还有一个条件，初始状态下2并不为空
        if (pRoot2 == NULL) return true; // 如果2为空，则说明2已经遍历结束还没遇到终止条件；此时1的状态如何无所谓
        if (pRoot1 == NULL) return false; // 2不为空而1为空，则说明1没有2大，显然不是子结构
        return (pRoot1->val==pRoot2->val) && isSubTree(pRoot1->left, pRoot2->left) 
            &&  isSubTree(pRoot1->right, pRoot2->right);
    }
};

• 运行时间：3ms
• 占用内存：460k