本文主要是介绍leetcode 951. 翻转等价二叉树,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
思想
题目中说了,只要经过一定次数的翻转操作后,能使二叉树 X 等于二叉树 Y,我们就称二叉树 X 翻转等价于二叉树 Y,所以不一定在每一个root都要进行翻转。
上面的树的颜色分别代表递归思路(3)的①②③情况。
- 标为黄色代表: r o o t 1 root_1 root1和 r o o t 2 root_2 root2相比
- 标为橘色代表以下两种情况都可以:
- l e f t 1 left_1 left1和 l e f t 2 left_2 left2相比, r i g h t 1 right_1 right1和 r i g h t 2 right_2 right2相比
- l e f t 1 left_1 left1和 r i g h t 2 right_2 right2相比, r i g h t 1 right_1 right1和 l e f t 2 left_2 left2相比
代码
/*** 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 flipEquiv(TreeNode* root1, TreeNode* root2) {if(!root1 && !root2)return true; // root1和root2都不存在,就说明当前节点相同bool flag = false;if(root1 && root2){if(root1->val == root2->val){ // root1和root2都存在,且值相同flag = true; // 就说明当前节点是相同的}return flag && (flipEquiv(root1->right, root2->left) && flipEquiv(root1->left, root2->right)) || (flipEquiv(root1->right, root2->right) && flipEquiv(root1->left, root2->left)); // root1->val == root2->val代表当前节点相不相同 && (flipEquiv(root1->right, root2->left) && flipEquiv(root1->left, root2->right))代表翻转后情况相不相同 // || (flipEquiv(root1->right, root2->right) && flipEquiv(root1->left, root2->left))代表不翻转情况相不相同}return false; // root1和root2其中有一个存在,另一个不存在就是false}
};
进一步观察我又进行了化简
class Solution {
public:bool flipEquiv(TreeNode* root1, TreeNode* root2) {if(!root1 && !root2)return true; // root1和root2都不存在,就说明当前节点相同if(root1 && root2){return root1->val == root2->val && (flipEquiv(root1->right, root2->left) && flipEquiv(root1->left, root2->right)) || (flipEquiv(root1->right, root2->right) && flipEquiv(root1->left, root2->left)); // root1->val == root2->val代表当前节点相不相同 && (flipEquiv(root1->right, root2->left) && flipEquiv(root1->left, root2->right))代表翻转后情况相不相同 // || (flipEquiv(root1->right, root2->right) && flipEquiv(root1->left, root2->left))代表不翻转情况相不相同}return false; // root1和root2其中有一个存在,另一个不存在就是false}
};
这篇关于leetcode 951. 翻转等价二叉树的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!