力扣打卡：429. N 叉樹的層序遍歷
解題思路
這道題目就是二叉樹的變形，二叉樹的左右子節點變為了一群子結點
- 寫出二叉樹的框架
- 原來 判斷添加 左右子節點的位置，改成 判斷添加 一群子節點
代碼
/*
// Definition for a Node.
class Node {
public int val;
public List<Node> children;
public Node() {}
public Node(int _val) {
val = _val;
}
public Node(int _val, List<Node> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public List<List<Integer>> levelOrder(Node root) {
// 這道題目也就是二叉樹的變形版本，left和right變為了_children，判斷children即可
List<List<Integer>> ans = new ArrayList<>();
if(root==null) return ans; // 判斷是否為空，因為即使為空，que也會添加
Queue<Node> que = new LinkedList<>();
que.offer(root);
while(!que.isEmpty()){
int size = que.size(); // 這個變數非常重要，因為記錄了同一層的個數
List<Integer> list = new ArrayList<>();
for(int i=0; i<size; i++){
Node node = que.poll(); // 將先進的節點彈出
list.add(node.val);
for(Node n : node.children) que.offer(n); // 原來添加左右子節點，改成了添加一群子節點
}
ans.add(list);
}
return ans;
}
}
如果是N叉樹的前中后呢？
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/350835.html
標籤：其他
上一篇：對稱二叉樹