LeetCode 572. 另一个树的子树（二叉树迭代器）

Michael阿明

发布于 2020-07-13 16:05:11

2660

发布于 2020-07-13 16:05:11

1. 题目

给定两个非空二叉树 s 和 t，检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。

示例 1:
给定的树 s:

     3
    / \
   4   5
  / \
 1   2
给定的树 t：

   4 
  / \
 1   2
返回 true，因为 t 与 s 的一个子树拥有相同的结构和节点值。

示例 2:
给定的树 s：

     3
    / \
   4   5
  / \
 1   2
    /
   0
给定的树 t：

   4
  / \
 1   2
返回 false。

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/subtree-of-another-tree 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 二叉树迭代器

对树s中的每个节点Si，Si与t进行递归比较
Si采用二叉树迭代器产生
该解法相当于暴力查找

class Solution {
	TreeNode *cur, *temp;
	stack<TreeNode*> stk;
public:
    bool isSubtree(TreeNode* s, TreeNode* t) {
		cur = s;
		bool ans = false;
		TreeNode *Si = next();//Si为二叉树的每个节点
		while(!ans && Si != NULL)//没有答案，Si还没遍历完
		{
			ans = isSub(Si,t);//对每个节点开始，跟t进行比较
			Si = next();
		}
    	return ans;
    }

    bool isSub(TreeNode *s, TreeNode *t)//比较Si与t的结构相同吗？
    {
    	if(!s && !t)
    		return true;
    	else if((!s && t) || (s && !t))
    		return false;
        if(s->val == t->val)//相等，还要继续比较子树
        	return isSub(s->left,t->left)&&isSub(s->right,t->right);
        return false;
    }

    TreeNode* next()//二叉树迭代器
    {
    	while(cur || !stk.empty())
    	{
    		while(cur)
    		{
    			stk.push(cur);
    			cur = cur->left;
    		}
    		temp = stk.top();
    		cur = stk.top()->right;
    		stk.pop();
    		return temp;
    	}
    	return NULL;
    }
};

别人的解法总是很优雅

class Solution {
public:
    bool isSubtree(TreeNode* s, TreeNode* t) {
        if (!s) 
        	return false;
        return isSubtree(s->left, t) || SameTree(s, t) || isSubtree(s->right, t);
    }

    bool SameTree(TreeNode* s, TreeNode* t) {
        if (!s && !t) 
        	return true;
        if (!s || !t) 
        	return false;
        return s->val == t->val && SameTree(s->left, t->left) && SameTree(s->right, t->right);
    }
};

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2019-11-16 ，如有侵权请联系 cloudcommunity@tencent.com 删除

二叉树