题目地址:
https://www.lintcode.com/problem/tweaked-identical-binary-tree/description
给定两棵二叉树,允许对其中一棵树的任意节点的左右子树进行对换。问其中一棵树能否变为另一棵(即问在允许交换左右子树的情况下两棵树是否等价)。题目保证树中的节点的值彼此不等。
思路是递归。如果两棵树都为空则返回true,若两树只有一棵为空则返回false,否则接下来先判断树根是否相等,然后再分两种情况:左左和右右,以及左右和右左。“左左”的意思是第一棵树的左子树和第二棵树的左子树是否等价,以此类推。代码如下:
public class Solution {
/**
* @param a: the root of binary tree a.
* @param b: the root of binary tree b.
* @return: true if they are tweaked identical, or false.
*/
public boolean isTweakedIdentical(TreeNode a, TreeNode b) {
// write your code here
if (a == null && b == null) {
return true;
}
if (a == null || b == null) {
return false;
}
return a.val == b.val &&
((isTweakedIdentical(a.left, b.left) && isTweakedIdentical(a.right, b.right))
|| (isTweakedIdentical(a.left, b.right) && isTweakedIdentical(a.right, b.left)));
}
}
class TreeNode {
int val;
TreeNode left, right;
public TreeNode(int val) {
this.val = val;
}
}
时间复杂度 O ( n ) O(n) O(n),空间 O ( h ) O(h) O(h)。
时间复杂度大概可以以此估计:
T
(
a
,
b
)
=
O
(
1
)
+
4
T
(
a
2
,
b
2
)
T(a,b)=O(1)+4T(\frac{a}{2},\frac{b}{2})
T(a,b)=O(1)+4T(2a,2b)由主定理知道
T
(
a
,
b
)
=
O
(
n
)
T(a,b)=O(n)
T(a,b)=O(n),其中
n
n
n代表两棵树的规模。