例题链接:二叉树的层序遍历
此文章用的是java实现,相对C语言,较为简单,如果使用C语言也可以做,只需要自己建立一个队列,但这样就比较耗时。
下面以这个二叉树为例,来展示思路(图文并茂)
一共有画了8张图,但是核心思路其实就是第1、2、3小点,完全不用担心复杂或者不好理解的哦
1.先把根节点,也就是root,进队列:
2.定义一个cur引用,用它来遍历整个二叉树。当队列不为空的时候,出队列。出队的元素就用cur来接收,把A结点中储存的元素记录到要返回的数组中:
3.检查A结点的左子子树和右子树是否为空,不为空,入队。注意,一定要按照先左后右的顺序,因为是层序遍历:
4.然后重复第二点的操作,出队一个结点,cur指向这个结点,把cur指向的那个结点所储存的元素放到要返回的数组中。
5.然后再重复第三步,看要不要把左右子节点,入队,不为空就入队:
6.然后再重复执行第二步,没错其实核心就是第1、2步,下面这几步都是循环执行的:
看吧,A.B.C ~ 遍历的层序顺序快出来了!(●'◡'●)
7.因为接下来的队列内节点都是叶子节点,所以第三步不会执行。
一直执行第二部,也就是一直出队,然后把节点元素放到ret中,就能得到这颗树的层序遍历了!
以上就是整体做题思路咯。
下面我们把他翻译成代码
事前准备:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ret = new ArrayList<>();//用来记录遍历顺序的,是个不定长二维数组
if (root == null) return ret;//当前节点是空,就直接返回ret(空)了
Queue<TreeNode> queue = new LinkedList<>();//借助这个队列来层序遍历
TreeNode cur = root;//用来遍历整个二叉树的节点
int size = 0;
queue.offer(cur);//队列先塞进头节点
}
}
这个size其实是用来记录每一层的节点数的:
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ret = new ArrayList<>();//用来记录遍历顺序的,是个不定长二维数组
if (root == null) return ret;//当前节点是空,就直接返回ret(空)了
Queue<TreeNode> queue = new LinkedList<>();//借助这个队列来层序遍历
TreeNode cur = root;//用来遍历整个二叉树的节点
int size = 0;
queue.offer(cur);//队列先塞进头节点
while (!queue.isEmpty()) {//队列中全部元素走完才出这个循环
List<Integer> list = new ArrayList<>();//把当前层的所有节点放到这里,然后再这次外层循环完了后,把list(Add)到要返回的二维数组中
size = queue.size();//这一层结点的个数
while (size > 0) {
cur=queue.poll();//用cur接收
list.add(cur.val);
if (cur.left != null) {//先左边,进入队列,然后右边在进入
queue.offer(cur.left);
}
if (cur.right != null) {//右边入队
queue.offer(cur.right);
}
size--;
}
ret.add(list);//到了这一步,list已经存储了当前层的所有结点,把它放到二维数组中
}
return ret;
}
}
完