LeetCode——655.输出二叉树

通过万岁！！！

题目：给你一个二叉树，将每一层都放到一个一位维数组中。但是这里有个要求，就是主要摆放的顺序。可以看一下B+数的数据接口，特别是叶子节点。其实这个题跟那个结构有些相似的，就是按照二叉树压扁以后，变成一个一维数组。每个节点在压扁的一维数组上面一定有一个下标。这个下面要与我们设计的那个每一层的一位数组的下标是对应的。每一层一个一维数组，最后返回的就是一个二维数组。
思路：其实这个题描述起来比较复杂，但是思路还是比较简单的。每一层一个一维数组，我们可以看根节点，那么他就是在这个数组的中间。所以左子树应该在根节点左边这块区间的中间，右子树在根节点右边这块区间的中间。一次我们就可以想到递归了。前提是我们需要知道每个数组的长度，已经树的深度。这个我用的层序遍历找到了深度，那么每个一维数组的长度也就知道。然后这里顺便总结一下递归，这里我们每一层都那排一个节点的位置，节点的位置是这个区间的中间，那么区间可以用两个变量传递进来。推出条件就是树的节点是空。最后就是这个数组是初始化好的，初始化的时候，所有值都是空字符串。
技巧：深度优先遍历

class Solution {
    public List<List<String>> printTree(TreeNode root) {
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int size;
        int numF = 0;
        while (!queue.isEmpty()) {
            size = queue.size();
            for (int i = 0; i < size; i++) {
                TreeNode remove = queue.remove();
                if (remove.left != null) {
                    queue.add(remove.left);
                }
                if (remove.right != null) {
                    queue.add(remove.right);
                }
            }
            numF++;
        }
        int len = (1 << numF) - 1;// 多少列
        List<List<String>> ans = new ArrayList<>();
        for (int i = 0; i < numF; i++) {
            List<String> temp = new ArrayList<>();
            for (int j = 0; j < len; j++) {
                temp.add("");
            }
            ans.add(temp);
        }
        fillAns(0, len, 0, ans, root);
        return ans;
    }


    private void fillAns(int begin, int end, int f, List<List<String>> ans, TreeNode root) {
        if (root == null) {
            return;
        }
        int idx = (begin + end) / 2;
        ans.get(f).set(idx, root.val + "");
        fillAns(begin, idx - 1, f + 1, ans, root.left);
        fillAns(idx + 1, end, f + 1, ans, root.right);
    }
}

总结：题目比较有意思的，也挺新颖的。这个递归的方式，我也是想了好久，并且这种方法的空间负责都不好。现在自己越来越懒了，哎。

原文地址：https://www.jb51.cc/wenti/3287011.html

LeetCode——655.输出二叉树

相关推荐