剑指Offer——树的子结构

Question

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

Solution

• 先序遍历第一棵树，找到和第二棵树根节点值相同的节点，然后判断两棵树是否一样， 最坏时间复杂度为O(n * m)

• 中序遍历+先序遍历、中序遍历+后序遍历，如果每个结点值都不相同的话，可以完全确定一颗树，如果值有相同的话，可以把空结点也考虑到遍历结果中，也可以唯一确定，如果结点值都是1的话，如果不考虑空结点的话，不同的树

• 判断第二棵树的遍历结果是不是第一棵树的子串，就相当于判断一个字符串是否为另外一个字符串的子串，可以考虑用KMP算法。 时间复杂度为O(4(m + n)) = O(m + n)

Code

/*
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;
    	
        bool isHas = false;
        
        if (pRoot1->val == pRoot2->val) {
            isHas = isSubtree(pRoot1, pRoot2);
        }
        
        if (!isHas)
        	 isHas = HasSubtree(pRoot1->left, pRoot2);
        if (!isHas)
             isHas = HasSubtree(pRoot1->right, pRoot2);
        
        return isHas;
    }
    bool isSubtree(TreeNode* p1, TreeNode* p2) {
        // 判断结构是否一样
		if (p2 == NULL)
            return true;
        // p2 有的节点，在p1没有
        if (p1 == NULL)
            return false;
        
        // 判断值是否一样
        if (p1->val != p2->val)
            return false;
        return isSubtree(p1->left, p2->left) && isSubtree(p1->right, p2->right);  
    }
};