100 相同的树

题目

给你两棵二叉树的根节点 p 和 q ，编写一个函数来检验这两棵树是否相同。

如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

示例 1：
输入：p = [1,2,3], q = [1,2,3]
输出：true

示例 2：
输入：p = [1,2], q = [1,null,2]
输出：false

示例 3：
输入：p = [1,2,1], q = [1,1,2]
输出：false

• 递归版

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool isSameTree(TreeNode* p, TreeNode* q) {
        if(p==NULL && q==NULL) return true;
        else if(p==NULL || q==NULL) return false;
        else return p->val == q->val && isSameTree(p->left,q->left) && isSameTree(p->right, q->right);
    }
};

• 时间复杂度O(n)

• 空间复杂度O(logn)

• 思路

  • 确定函数的参数和返回值
    • 判断是否相同故返回值为bool，传入参数为两棵树的根节点
  • 确定终止条件
    • 如果两个节点均为空，返回true
    • 如果一个为空，一个不为空，返回false
  • 确定单层递归调用的逻辑
    • 如果两个都不为空，返回它们值是否相同，相同则递归调用判断他们的孩子

• 迭代版

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    bool isSameTree(TreeNode* p, TreeNode* q) {
        stack<TreeNode*> stk;
        stk.push(p);
        stk.push(q);
        while(!stk.empty()){
        	//判断当前栈顶节点和它的下一个节点是否一致
            p = stk.top(); stk.pop();
            q = stk.top(); stk.pop();
            if(p==NULL && q==NULL) continue;
            if(p==NULL || q==NULL) return false;
            if(p->val != q->val) return false;
            //将其右孩子入栈
            stk.push(p->right);
            stk.push(q->right);
            //将其左孩子入栈
            stk.push(p->left);
            stk.push(q->left);
        }
        return true;
    }
};

• 时间复杂度O(n)
• 空间复杂度O(logn)
• 思路
  • 用栈模拟递归实现判断
  • 先将两棵树的根节点相继入栈，在栈非空的条件下依次判断两棵树中对应位置的节点是否一致。
    • 记录此时栈顶的两个元素，弹栈后，判断是否一致。
    • 入栈两棵树中各自位置该节点的左孩子，入栈对应位置的右孩子。保证栈中相邻的两个节点分别为两棵树的同一个位置，每次判断前弹栈两次。

递归参考详解