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;
}
};