我们可以为二叉树 T 定义一个翻转操作,如下所示:选择任意节点,然后交换它的左子树和右子树。
只要经过一定次数的翻转操作后,能使 X 等于 Y,我们就称二叉树 X 翻转等价于二叉树 Y。
编写一个判断两个二叉树是否是翻转等价的函数。这些树由根节点 root1
和 root2
给出。
示例:
输入:root1 = [1,2,3,4,5,6,null,null,null,7,8], root2 = [1,3,2,null,6,4,5,null,null,null,null,8,7]
输出:true
解释:We flipped at nodes with values 1, 3, and 5.
提示:
- 每棵树最多有
100
个节点。 - 每棵树中的每个值都是唯一的、在
[0, 99]
范围内的整数。
解答:
最后时间不够了,再加上进入思维陷阱,没写出来。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public boolean flipEquiv(TreeNode root1, TreeNode root2) {
// 终止条件
if (root1 == null && root2 == null) return true;
if (root2 == null || root1 == null || (root1.val != root2.val)) {
return false;
}
// 递归调用
return (flipEquiv(root1.right, root2.right) && flipEquiv(root1.left, root2.left)) ||
(flipEquiv(root1.left, root2.right) && flipEquiv(root1.right, root2.left));
}
}
规格化遍历:
class Solution {
public boolean flipEquiv(TreeNode root1, TreeNode root2) {
// 用链表记录遍历结果
List<Integer> list1 = new ArrayList<>();
List<Integer> list2 = new ArrayList<>();
dfs(root1, list1);
dfs(root2, list2);
return list1.equals(list2);
}
public void dfs(TreeNode root, List list) {
if (root != null) list.add(root.val);
// 通过填充-1来防止不同树的遍历结果相同,如{1,2,3}与{1,null,2,null,null,3,null}
else {
list.add(-1);
return;
}
int L = root.left != null ? root.left.val : -1;
int R = root.right != null ? root.right.val : -1;
// 按照左右子树的大小顺序遍历子节点
if (R > L) {
dfs(root.right, list);
dfs(root.left, list);
} else {
dfs(root.left, list);
dfs(root.right, list);
}
}
}