对称二叉树&相同的树&另一棵树的子树&树的子结构&合并二叉树

对称的二叉树

请实现一个函数,用来判断一棵二叉树是不是对称的。如果一棵二叉树和它的镜像一样,那么它是对称的。

例如,二叉树 [1,2,2,3,4,4,3] 是对称的。
  1
   / \
  2   2
 / \ / \
3  4 4  3
但是下面这个 [1,2,2,null,3,null,3] 则不是镜像对称的:
    1
   / \
  2   2
   \   \
   3    3

示例 1:

输入:root = [1,2,2,3,4,4,3]
输出:true

示例 2:

输入:root = [1,2,2,null,3,null,3]
输出:false

递归

public boolean isSymmetric(TreeNode root) {
    if(root==null){
        return true;
    }    
    return helper(root.left,root.right);    
}

public boolean helper(TreeNode root1,TreeNode root2){
    if(root1==null&&root2==null){
        return true;
    }
    if(root1==null||root2==null){
        return false;
    }
    if(root1.val==root2.val){
        return helper(root1.left,root2.right)&&helper(root1.right,root2.left);
    }
    return false;
}

迭代
使用栈或队列,两两一对访问

public boolean isSymmetric(TreeNode root) {
    if (root == null) {
        return true;
    }
    Stack stack = new Stack<>();
    stack.push(root.left);
    stack.push(root.right);
    while (!stack.isEmpty()) {
        TreeNode node1 = stack.pop();
        TreeNode node2 = stack.pop();
        if (node1 == null && node2 == null) {
            continue;
        }
        if (node1 == null || node2 == null) {
            return false;
        }
        if (node1.val != node2.val) {
            return false;
        }
        stack.push(node1.left);
        stack.push(node2.right);
        stack.push(node1.right);
        stack.push(node2.left);
    }
    return true;
}

相同的树

给你两棵二叉树的根节点 pq ,编写一个函数来检验这两棵树是否相同。
如果两个树在结构上相同,并且节点具有相同的值,则认为它们是相同的。

示例 1:

输入:p = [1,2,3], q = [1,2,3]
输出:true

递归

public boolean isSameTree(TreeNode p, TreeNode q) {
    if (p == null && q == null) {
        return true;
    }
    if (p == null || q == null) {
        return false;
    }
    if (p.val != q.val) {
        return false;
    }
    return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
}

迭代

public boolean isSameTree(TreeNode p, TreeNode q) {
    Stack stack = new Stack<>();
    stack.push(p);
    stack.push(q);
    while (!stack.isEmpty()) {
        TreeNode node1 = stack.pop();
        TreeNode node2 = stack.pop();
        if (node1 == null && node2 == null) {
            continue;
        }
        if (node1 == null || node2 == null) {
            return false;
        }
        if (node1.val != node2.val) {
            return false;
        }
        stack.push(node1.left);
        stack.push(node2.left);
        stack.push(node1.right);
        stack.push(node2.right);
    }
    return true;
}

另一棵树的子树

给你两棵二叉树 rootsubRoot 。检验 root 中是否包含和 subRoot 具有相同结构和节点值的子树。如果存在,返回 true ;否则,返回 false

二叉树 tree 的一棵子树包括 tree 的某个节点和这个节点的所有后代节点。tree 也可以看做它自身的一棵子树。

示例 1:

输入:root = [3,4,5,1,2], subRoot = [4,1,2]
输出:true

方一:两层递归
用主树的每一个节点做根与子树比较是否匹配

public boolean isSubtree(TreeNode root, TreeNode subRoot) {
    if (root == null) {
        return subRoot == null;
    }
    return isSameTree(root, subRoot) || isSubtree(root.left, subRoot) || isSubtree(root.right, subRoot);
}

public boolean isSameTree(TreeNode p, TreeNode q) {
    if (p == null && q == null) {
        return true;
    }
    if (p == null || q == null) {
        return false;
    }
    if (p.val != q.val) {
        return false;
    }
    return isSameTree(p.left, q.left) && isSameTree(p.right, q.right);
}

时间复杂度:对于每一个 s 上的点,都需要做一次深度优先搜索来和 t 匹配,匹配一次的时间代价是 O(∣t∣),那么总的时间代价就是 O(∣s∣×∣t∣)

方法二:转字符串比较
遍历两棵树得到字符串,注意需要将空节点算上

public boolean isSubtree(TreeNode root, TreeNode subRoot) {
    StringBuilder sb1 = new StringBuilder();
    StringBuilder sb2 = new StringBuilder();
    dfs(root, sb1);
    dfs(subRoot, sb2);
    return sb1.toString().indexOf(sb2.toString()) != -1;
}

private void dfs(TreeNode root, StringBuilder sb) {
    if (root == null) {
        sb.append("#").append("null");
    } else {
        sb.append("#").append(root.val);
        dfs(root.left, sb);
        dfs(root.right, sb);
    }
}

时间复杂度:遍历两棵树得到深度优先搜索序列的时间代价是)O(∣s∣+∣t∣),在匹配的时候,如果使用暴力匹配,时间代价为 O(∣s∣×∣t∣),使用 KMP 进行串匹配的时间代价都是O(∣s∣+∣t∣)

树的子结构

输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。

例如:

给定的树 A:
     3
    / \
   4   5
  / \
 1   2
给定的树 B:
   4 
  /
 1
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值

示例 1:

输入:A = [1,2,3], B = [3,1]
输出:false

示例 2:

输入:A = [3,4,5,1,2], B = [4,1]
输出:true

递归

public boolean isSubStructure(TreeNode A, TreeNode B) {
    if (B == null || A == null) {
        return false;
    }
    return dfs(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
}

public boolean dfs(TreeNode A, TreeNode B) {
    if (B == null) {
        return true;
    }
    if (A == null) {
        return false;
    }
    if (A.val != B.val) {
        return false;
    }
    return dfs(A.left, B.left) && dfs(A.right, B.right);
}

迭代

public boolean isSubStructure(TreeNode A, TreeNode B) {
    if (B == null || A == null) {
        return false;
    }
    return helper(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
}

public boolean helper(TreeNode A, TreeNode B) {
    Stack stack = new Stack<>();
    stack.push(A);
    stack.push(B);
    while (!stack.isEmpty()) {
        TreeNode node2 = stack.pop();
        TreeNode node1 = stack.pop();
        if (node1 == null || node1.val != node2.val) {
            return false;
        }
        // 相比于比较两棵树是否相等,加了if判断
        if (node2.left != null) {
            stack.push(node1.left);
            stack.push(node2.left);
        }
        if (node2.right != null) {
            stack.push(node1.right);
            stack.push(node2.right);
        }
    }
    return true;
}

合并二叉树

给定两个二叉树,想象当你将它们中的一个覆盖到另一个上时,两个二叉树的一些节点便会重叠。

你需要将他们合并为一个新的二叉树。合并的规则是如果两个节点重叠,那么将他们的值相加作为节点合并后的新值,否则**不为 **NULL 的节点将直接作为新二叉树的节点。

示例 1:

输入: 
    Tree 1                     Tree 2                  
          1                         2                             
         / \                       / \                            
        3   2                     1   3                        
       /                           \   \                      
      5                             4   7                  
输出: 
合并后的树:
         3
        / \
       4   5
      / \   \ 
     5   4   7

注意: 合并必须从两个树的根节点开始。

递归
对这两棵树同时进行前序遍历,并将对应的节点进行合并。在遍历时,如果两棵树的当前节点均不为空,我们就将它们的值进行相加,并对它们的左孩子和右孩子进行递归合并;如果其中有一棵树为空,那么我们返回另一颗树作为结果;如果两棵树均为空,此时返回任意一棵树均可(因为都是空)。

public TreeNode mergeTrees(TreeNode t1, TreeNode t2) {
    if (t1 == null) {
        return t2;
    }
    if (t2 == null) {
        return t1;
    }
    t1.val += t2.val;
    t1.left = mergeTrees(t1.left, t2.left);
    t1.right = mergeTrees(t1.right, t2.right);
    return t1;
}

迭代

public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
    if (root1 == null) {
        return root2;
    }
    if (root2 == null) {
        return root1;
    }
    Stack stack = new Stack<>();
    stack.add(root1);
    stack.add(root2);
    while (!stack.isEmpty()) {
        TreeNode node2 = stack.pop();
        TreeNode node1 = stack.pop();
        node1.val += node2.val;
        // 如果两棵树的左子树不为空,将其左子树入栈
        // 如果root1左子树为空,将root2的左子树作为root1的左子树
        // 如果root2左子树为空,合并之后只有root1的左子树,即不做处理
        if (node1.left != null && node2.left != null) {
            stack.add(node1.left);
            stack.add(node2.left);
        } else if (node1.left == null) {
            node1.left = node2.left;
        }
        if (node1.right != null && node2.right != null) {
            stack.add(node1.right);
            stack.add(node2.right);
        } else if (node1.right == null) {
            node1.right = node2.right;
        }
    }
    return root1;
}

你可能感兴趣的:(对称二叉树&相同的树&另一棵树的子树&树的子结构&合并二叉树)