例题链接:二叉树的层序遍历

此文章用的是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;
    }
}

完

Logo

北京人形旗下天工造物具身智能开源社区,聚焦具身天工与慧思开物两大平台

更多推荐