LeetCode #572 Subtree of Another tree 树的遍历

Description　　

---

 

Given two non-empty binary trees s and t, check whether tree t has exactly the same structure and node values with a subtree of s. A subtree of s is a tree consists of a node in s and all of this node's descendants. The tree s could also be considered as a subtree of itself.

Example 1:
Given tree s:

     3
    / \
   4   5
  / \
 1   2

Given tree t:

   4 
  / \
 1   2

Return true, because t has the same structure and node values with a subtree of s.

 

Example 2:
Given tree s:

     3
    / \
   4   5
  / \
 1   2
    /
   0

Given tree t:

   4
  / \
 1   2

Return false.

 

思路

---

 

　　判断一棵树是否是另外一棵树的子树，很直接的一道题，但是卡了我一个小时，还是太弱了。

　　这道题重点在于递归搜索时如果匹配失败了要如何回溯，一开始我是这么写的：

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

　　过不了“ [1, 1, NULL]，[1] ”这组数据，因为可以很明显地发现我的搜索是没有回溯的，一但匹配上就自动向下继续匹配直到失败，而没有回溯并搜索s的子树。

　　为了解决回溯搜索问题，我通过增加一个方法 isSame 去判断两棵树是否相同，这样的好处是模拟了回溯，当匹配失效时还能够返回到 isSubtree 继续匹配s的子树。

　　AC代码如下：

#include<iostream>
#include<algorithm>

struct TreeNode{
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution{
public:
    bool isSubtree(TreeNode *s, TreeNode *t) {
        if (!s) return false; //防止越界

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

        if (s->val != t->val) return false;
        return isSame(s->left, t->left) && isSame(s->right, t->right);
    }
};