【LeetCode-中等】剑指 Offer 26. 树的子结构（C++实现）

题目链接

剑指 Offer 26. 树的子结构

实现思路

对 A 进行 DFS，沿途进行判断 B 是否是 A 的子结构。

• 如果 A 为空或者 B 为空，根据题目的意思，返回 false。

• 否则，对 A 进行 DFS。

  if (A == NULL) {
      return false;
  }
  bool ans = isSubTree(A, B);
  if (ans) {
      return true;
  } // ans == false
  return traversal(A->left, B) || traversal(A->right, B); 
  

  深搜过程中，沿途对 A、B 进行判断（先暂时定义为 bool isSubTree() ），如果当前节点已经满足条件，就不用再往下搜索，直接返回 true；

  如果还不满足，则对它的左右孩子进行搜索，左右孩子中，只要其中的一个满足条件即可，所以使用 || 。

• 最后是bool isSubTree()。

  • 如果 A、B 同时为空，返回 true；
  • 否则，此时至少有一个节点不为空：
    • 有一个节点为空：
      • 如果此时 A 为空，则返回 false（说明 B 不是 A 的子结构）。
      • 如果此时 B 为空，因为 B 只需要满足是 A 的一部分这个条件，所以返回 true。
    • A、B都不为空：对它们的值进行判断，不一样则返回false，否则继续对A、B的左右孩子进行递归判断。

实现代码（C++）

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isSubTree(TreeNode *A, TreeNode *B) {
        if (A == NULL && B == NULL) { 
            return true;
        }
        if (A == NULL) { // A 为空 返回false
            return false;
        }
        if (B == NULL) { // B 为空，判断到B的一个叶子结点之外，返回true
            return true;
        }
        // A != NULL && B != NULL
        if (A->val != B->val) {
            return false;
        }
        return isSubTree(A->left, B->left) && isSubTree(A->right, B->right);
    }

    bool traversal(TreeNode* A, TreeNode* B) {
        if (A == NULL) {
            return false;
        }
        bool ans = isSubTree(A, B);
        if (ans) {
            return true;
        } // ans == false
        return traversal(A->left, B) || traversal(A->right, B); 
    }

    bool isSubStructure(TreeNode* A, TreeNode* B) {
        if (A == NULL || B == NULL) {
            return false;
        }
        return traversal(A, B);
    }
};