问题描述
问题分析
分析题目,使用BFS和DFS都可以,这里使用DFS:
- 首先构建DFS(当前节点root,层次level, 树形List, 数值List)。BFS分树形List和数值List的原因是,按照树形List的父子结构进行DFS,然后把每个节点的val存入数值List中,作为结果返回。
- DFS的思想:自顶向下,依次对每一个节点进行DFS,能进就进,不能进就退,然后选择兄弟节点进行DFS。
- 最后将数值List逆置即可。
解法:DFS
- 时间复杂度:O( n ),其中n表示节点个数。
Java代码
package com.company;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
public class Main {
static public class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
public static void main(String[] args) {
TreeNode root = new TreeNode(3);
root.right = new TreeNode(20);
root.left = new TreeNode(9);
root.right.left = new TreeNode(15);
root.right.right = new TreeNode(7);
List<List<Integer>> reslt = levelOrderBottom(root);
System.out.println(111);
}
static public List<List<Integer>> levelOrderBottom(TreeNode root) {
if (root == null){
return new ArrayList<List<Integer>>();
}
List<List<Integer>> result = new ArrayList<List<Integer>>();
List<List<TreeNode>> trees = new ArrayList<List<TreeNode>>();
DFS(root, 0,trees, result);
//reverse
Collections.reverse(result);
return result;
}
static public void DFS(TreeNode root, int level,List<List<TreeNode>> trees, List<List<Integer>> result) {
if (trees.size() <= level){
//add myself
List<TreeNode> tree = new ArrayList<TreeNode>();
tree.add(root);
trees.add(tree);
List<Integer> list = new ArrayList<Integer>();
list.add(root.val);
result.add(list);
}else {
//add myself
List<TreeNode> tree = trees.get(level);
tree.add(root);
trees.set(level, tree);
List<Integer> list = result.get(level);
list.add(root.val);
result.set(level, list);
}
//add my children
if (root.left != null){
DFS(root.left, level + 1, trees, result);
}
if (root.right != null){
DFS(root.right, level + 1, trees, result);
}
}
}
结果分析
以上代码的执行结果:
执行时间 | 内存消耗 |
---|---|
1 ms | 36.4 MB |