一、题目
给定一个二叉树,返回该二叉树层序遍历的结果,(从左到右,一层一层地遍历)
例如:
给定的二叉树是{3,9,20,null,null,15,7},
该二叉树层序遍历的结果是
[[3],[9,20],[15,7]]
二、解决方案
2.0 树节点构造
public class TreeNode {int val = 0;TreeNode left = null;TreeNode right = null;}
2.1 直接逐层遍历,定义nextNodeList和curLevelList,每遍历一层,就将当前层节点curList添加到结果res中。
// 定义nextNodeList和curLevelList,每遍历一层,就将当前层节点curList添加到结果res中public List<List<Integer>> levelOrderOne(TreeNode root) {List<List<Integer>> res = new ArrayList<>();if(root == null){return null;}// 存储当前层的节点List<TreeNode> curList = new ArrayList<>();curList.add(root);// 每遍历一层,就将当前层节点curList添加到结果res中while(!curList.isEmpty()){List<TreeNode> nextNodeList = new ArrayList<>();List<Integer> curLevelList = new ArrayList<>();for(TreeNode node : curList){curLevelList.add(node.val);if(node.left != null){nextNodeList.add(node.left);}if(node.right != null){nextNodeList.add(node.right);}}res.add(curLevelList);curList = nextNodeList;}return res;}
2.2 利用队列实现
因为队列是先进先出的数据结构,如果从左到右访问完一行节点,并在访问的时候把它们的子节点依次加入到队列,这时候它们的子节点也是按照顺序来进行的。且排在本行节点的后面,因此队列中出现的顺序正好也是从左到右,正好符合层次遍历的特点。
/*** 建立辅助队列,根节点首先进入队列。不管层次怎么访问,根节点一定是第一个,那它肯定排在队伍的最前面。* 每次进入一层,统计队列的元素个数,因为每当访问完一层,下一层作为这一层的子节点,一定都加入队列,而再下一层还没有加入,因此此时队列中的元素个数就是这一层的元素个数。* 每次遍历这一层这么多的节点数,将其依次从队列中弹出,然后加入这一行的一维数组中,如果它们有子节点,依次加入队列排队等待访问。* 访问完这一层的元素后,将这个一维数组加入二维数组中,再访问下一层。*/public List<List<Integer>> levelOrderTwo(TreeNode root) {List<List<Integer>> res = new ArrayList<>();if(root == null){return null;}Queue<TreeNode> queue = new ArrayDeque<>();queue.add(root);while(!queue.isEmpty()){// 记录二叉树的某一行List<Integer> currentRow = new ArrayList();int currentQueueSize = queue.size();for (int i = 0; i < currentQueueSize; i++) {// 先存当前根节点TreeNode node = queue.poll();currentRow.add(node.val);// 若是左右孩子存在,则存入左右孩子作为下一个层次if(node.left!= null){queue.add(node.left);}if(node.right!=null){queue.add(node.right);}}res.add(currentRow);}return res;}