二叉树常见题型总结

1.将二叉搜索树变平衡

关键点：二叉搜索树的中序遍历是递增的，所以题目转换为将二叉搜索树转换为递增数组，再将递增数组转换为平衡二叉树

class Solution {
    List<Integer> list = new ArrayList<>();
    public TreeNode balanceBST(TreeNode root) {
        transform(root);
        return getTree(0, list.size() - 1, list);
    }

    // 将递增集合转换为平衡二叉树
    public TreeNode getTree(int left, int right, List<Integer> list) {
        if (right < left) {
            return null;
        }
        int mid = (right + left) / 2;
        TreeNode root = new TreeNode(list.get(mid));
        root.left = getTree(left, mid - 1, list);
        root.right = getTree(mid + 1, right, list);
        return root;
    }

    // 将二叉搜索树添加到链表
    public void transform(TreeNode cur) {
        if (cur == null) {
            return;
        }
        transform(cur.left);
        list.add(cur.val);
        transform(cur.right);
    }
}

2.填充每个节点的下一个右侧节点指针

因为是完美二叉树

同一个父节点的左侧节点的下一个右侧节点就是父节点的右侧节点

而右侧节点的下一个右侧节点是父节点的下一个右侧节点的左节点

class Solution {
    public Node connect(Node root) {
        dG(root);
        return root;        
    }

    public void dG(Node cur) {
        if (cur == null) {
            return;
        }
        if (cur.left != null) {
            cur.left.next = cur.right;
        }
        if (cur.right != null) {
            if (cur.next != null) {
                cur.right.next = cur.next.left;
            } else {
                cur.right.next = null;
            }
        }

        dG(cur.left);
        dG(cur.right);
    }
}

3.相同的树

给定两个二叉树，编写一个函数来检验它们是否相同。

如果两个树在结构上相同，并且节点具有相同的值，则认为它们是相同的。

如果比较的两个节点都为空，则为相同的，如果两个节点有一个为空，或者值不相等，则为不相同

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