和我一起迎接明天的太阳吧

klaus08

焦虑源于行动的匮乏

JZ58 对称的二叉树

原题链接


描述

请实现一个函数,用来判断一棵二叉树是不是对称的。注意,如果一个二叉树同此二叉树的镜像是同样的,定义其为对称的。


示例

输入:{8,6,6,5,7,7,5}
返回值:true

思路

本来是想把二叉树层次遍历,每一层保存在一个数组中,但是由于有空结点,层次遍历时就没办法把他加进去。

还有想法是中序+后续遍历结果如果相同会不会就是对称的?

目前的思路是从根节点开始深搜,判断 L == R && L.left == R.right && L.left == R.right。


解答

import java.util.ArrayDeque;
import java.util.ArrayList;
import java.util.Queue;

public class Solution {
    public boolean isSymmetrical(TreeNode pRoot) {
        if (pRoot == null) return true;
        return check(pRoot.left, pRoot.right);
    }

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

有bug的,用MAX_VALUE保存null结点。

class Solution {
    public boolean isSymmetric(TreeNode root) {
        if (root == null) return true;
        ArrayDeque<TreeNode> deque = new ArrayDeque<>();
        ArrayList<Integer> arrayList;
        deque.add(root);
        while (!deque.isEmpty()) {
            arrayList = new ArrayList<>();
            int cnt = deque.size();
            while (cnt-- > 0) {
                TreeNode remove = deque.poll();
                arrayList.add(remove.val);
                // 判断当前结点不是填充的空结点
                if (remove.val != Integer.MAX_VALUE) {
                    deque.add(remove.left == null ? new TreeNode(Integer.MAX_VALUE) : remove.left);
                    deque.add(remove.right == null ? new TreeNode(Integer.MAX_VALUE) : remove.right);
                }
            }
            for (int i = 0; i < arrayList.size() / 2 + 1; i++) {
                                if (!Objects.equals(arrayList.get(i), arrayList.get(arrayList.size() - 1 - i)))

                    return false;
            }
        }

        return true;
    }
}
posted @ 2021-09-03 00:08  klaus08  阅读(27)  评论(0编辑  收藏  举报