广度优先算法 LeetCode102 题解

广度优先算法模板

def BFS(graph, start, end)：
    queue  = []
    queue.append([start]）
    visited. add(start)

	while queue:
	    node=quede.pop()
	    visited.add(node)
	    process(node）
	    nodes = generate_related_nodes(node)
	    queue. push(nodes)

给你一个二叉树，请你返回其按 层序遍历 得到的节点值。 （即逐层地，从左到右访问所有节点）。

示例：
二叉树：[3,9,20,null,null,15,7],

3

/
9 20
/
15 7

返回其层次遍历结果：

[
[3],
[9,20],
[15,7]
]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-tree-level-order-traversal
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
AC 代码

public class LeetCode102 {
    
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> result = new ArrayList<List<Integer>>();
        if (root == null)
            return result;
    
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        List<Integer> list1 = new ArrayList<Integer>();
        list1.add(root.val);
        result.add(list1);
        queue.add(root);
        while (!queue.isEmpty()) {
            List<Integer> list = new ArrayList<Integer>();
            // 每次将一层内容加到队列中，遍历这一层就可以了
            int len = queue.size();
            for(int i = 0;i < len;i++){
                TreeNode node = queue.poll();
                if (node.left != null) {
                    list.add(node.left.val);
                    queue.add(node.left);
                }
                if (node.right != null) {
                    list.add(node.right.val);
                    queue.add(node.right);
                }
            }
            if(!list.isEmpty()){
               result.add(list);
            }
        }
        return result;
    }

    /**
     * @param args
     */
    public static void main(String[] args) {
        
    }

}