通过万岁!!!
- 题目:给你一个二叉树,将每一层都放到一个一位维数组中。但是这里有个要求,就是主要摆放的顺序。可以看一下B+数的数据接口,特别是叶子节点。其实这个题跟那个结构有些相似的,就是按照二叉树压扁以后,变成一个一维数组。每个节点在压扁的一维数组上面一定有一个下标。这个下面要与我们设计的那个每一层的一位数组的下标是对应的。每一层一个一维数组,最后返回的就是一个二维数组。
- 思路:其实这个题描述起来比较复杂,但是思路还是比较简单的。每一层一个一维数组,我们可以看根节点,那么他就是在这个数组的中间。所以左子树应该在根节点左边这块区间的中间,右子树在根节点右边这块区间的中间。一次我们就可以想到递归了。前提是我们需要知道每个数组的长度,已经树的深度。这个我用的层序遍历找到了深度,那么每个一维数组的长度也就知道。然后这里顺便总结一下递归,这里我们每一层都那排一个节点的位置,节点的位置是这个区间的中间,那么区间可以用两个变量传递进来。推出条件就是树的节点是空。最后就是这个数组是初始化好的,初始化的时候,所有值都是空字符串。
- 技巧:深度优先遍历
java代码
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);
}
}
- 总结:题目比较有意思的,也挺新颖的。这个递归的方式,我也是想了好久,并且这种方法的空间负责都不好。现在自己越来越懒了,哎。