[算法] 二叉树的层级遍历

二叉树的层级遍历

解题思路：队列（先进先出）

• 先将根结点 add插入到队列中。
• 开启循环，poll取出头部结点，将该结点的左右孩子 add插入到队列尾部。
• 每次循环为二叉树的一层。用左右孩子结点替换父亲结点。
• 待队列为空，循环截止，遍历完整个二叉树。

Tals is cheap, Show me the code.

• Java Code

    class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode(int val) {
            this.val = val;
        }
    }

	// [BFS] 二叉树的层级遍历 
    private List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) return res;
        Queue<TreeNode> list = new LinkedList<>();
        list.add(root);
        while (!list.isEmpty()) {
            List<Integer> curLevel = new ArrayList<>();
            for (int i = 0; i < list.size(); i++) {
                TreeNode curNode = list.poll();// 遍历集合取出第一个结点
                curLevel.add(curNode.val);// 将结点值add到当前层。
                // 添加左右孩子结点到队列尾部
                if (curNode.left != null)
                    list.add(curNode.left);
                if (curNode.right != null)
                    list.add(curNode.right);
            }
            res.add(curLevel);
        }
        return res;
    }

• Kotlin Code

	class TreeNode(var `val`: Int) {
    	var leftNode: TreeNode? = null
    	var rightNode: TreeNode? = null
	}

	// [BFS] 二叉树的层级遍历 
    private fun levelOrder(root: TreeNode?): ArrayList<ArrayList<Int>> {
        val ret = arrayListOf<ArrayList<Int>>()
        if (root == null) return ret
        val list: Queue<TreeNode> = LinkedList<TreeNode>()
        list.add(root)
        while (!list.isEmpty()) {
            val curLevelData = arrayListOf<Int>()
            for (i in 0 until list.size) {
                val curNode = list.poll()
                curLevelData.add(curNode.`val`)
                if (curNode.leftNode != null)
                    list.add(curNode.leftNode!!)
                if (curNode.rightNode != null)
                    list.add(curNode.rightNode!!)
            }
            ret.add(curLevelData)
        }
        return ret
    }