【Lintcode】470. Tweaked Identical Binary Tree

题目地址:

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代表两棵树的规模。

你可能感兴趣的:(#,树,分治与堆)