Sicilly 等价二叉树 | 算法期末机考模拟题-CSDN博客

本文链接：https://blog.csdn.net/Sician_Lee/article/details/74607104

Description

两个二叉树结构相同，且对应结点的值相同，我们称这两个二叉树等价.

例如：以下两个二叉树等价
1 1
/ \ / \
2 3 2 3
而以下两个则不等价
1 1
/ \ / \
2 3 3 2
以下两个也不等价
1 1
/ \ / \
2 3 2 2

给出两个二叉树p和q，判断它们是否等价.

p和q的结点数不多于100000，每个结点的数值在1和1000000000之间.

请为下面的Solution类实现解决上述问题的isEqual函数，函数的两个参数p和q分别代表两个二叉树的根节点，如果以p和q为根的二叉树等价则函数返回true，否则返回false.

/**
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 isEqual(TreeNode* p, TreeNode* q) {
}
};

Thinking

此题简单地考查DFS，通过stack对两棵树DFS，同时比较两棵树对应各节点的值是否相等，还需要比较左右子树的结构（即是否同时存在相同的子树），一旦不同即返回false。

class Solution {
public:
    bool isEqual(TreeNode* p, TreeNode* q) {
        if(p == NULL && q == NULL) return true;
        if((p == NULL && q != NULL) || (p != NULL && q == NULL)) return false;
        stack<TreeNode*> s1, s2;
        TreeNode * temp1 = p;
        TreeNode * temp2 = q;
        s1.push(p); s2.push(q);
        while(!s1.empty() && !s2.empty()){
            temp1 = s1.top(); temp2 = s2.top(); 
            if(temp1->val != temp2->val) return false;
            s1.pop(); s2.pop();
            if(temp1 -> right && temp2 -> right){
                s1.push(temp1->right); 
                s2.push(temp2->right);
            }
            else if((temp1 -> right == NULL && temp2 -> right != NULL) || (temp1 -> right != NULL && temp2 -> right == NULL)) return false;
            if(temp1 -> left && temp2 -> left){
                s1.push(temp1->left); 
                s2.push(temp2->left);
            }
            else if((temp1 -> left == NULL && temp2 -> left != NULL) || (temp1 -> left != NULL && temp2 -> left == NULL)) return false;
        }
        //if(s1.empty() ^ s2.empty()) return false;
        return true;
    }
};