해결 방법 1:
너비 우선을 사용하여 레이어별로 트리를 순회하고 각 레이어의 노드를 먼저 스택에 넣은 다음 현재 스택의 크기인 이 레이어의 노드 수를 기록하고 각 레이어를 추가합니다. 해당 레이어의 항목을 List에 넣은 후 pop 작업을 수행하며, 각 레이어의 노드 수만큼 pop 작업을 수행합니다. 스택이 팝될 때마다 해당 하위 노드는 스택에 배치되고 스택이 비어 있고 순회가 완료될 때까지 해당 레이어의 해당 목록에 배치됩니다.
public List<List<Integer>> levelOrder(TreeNode root) {
Queue<TreeNode> queue=new LinkedList<>();
List<List<Integer>> lists=new LinkedList<>();
if (root==null){
return lists;
}
//offer 与 add ,add会在栈满的情况下报异常,而offer只会返回false
queue.offer(root);
while (!queue.isEmpty()){
List<Integer> list=new LinkedList<>();
int lenth=queue.size();
for (int i=0;i<lenth;i++){
TreeNode treeNode=queue.poll();
list.add(treeNode.val);
if (treeNode.left!=null){
queue.offer(treeNode.left);
}
if (treeNode.right!=null){
queue.offer(treeNode.right);
}
}
lists.add(list);
}
return lists;
}
해결 방법 2 :
깊이 우선순위를 사용하여 각 분기에 액세스한 후에만 다른 분기에 액세스합니다. 매번 방문하는 노드의 레이어 수를 기록하는 것이 핵심인데, 결과의 크기와 같다면 새로운 레이어를 입력했다는 뜻이고, 노드를 저장하려면 새 List를 생성해야 한다. 이 레이어의. 서로 다른 경우 결과에 이 레이어를 저장하는 목록을 꺼내 노드에 넣은 다음 결과에 다시 넣으세요. 노드가 빌 때까지 계속 아래로 순회하고 결과를 반환합니다.
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> lists=new LinkedList<>();
int level=0;
return dfs(root,lists,level);
}
public List<List<Integer>> dfs (TreeNode root,List<List<Integer>> result, int level){
if (root==null){
return result;
}
List<Integer> list =null;
if (result.size()==level){
list=new LinkedList<>();
list.add(root.val);
result.add(list);
}else {
list=result.get(level);
list.add(root.val);
result.set(level,list);//存进去了后还是要放回去的
}
dfs(root.left,result,level+1);
dfs(root.right,result,level+1);
return result;
}