廣度優先算法 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) {
        
    }

}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章