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