leetcode *102. 두 갈래 나무의 층차 훑어보기 & 검지 Offer 32 - II.위에서 아래로 두 갈래 나무 인쇄 II

*102.두 갈래 나무의 층차 훑어보기 & 검지 Offer 32 - II.위에서 아래로 두 갈래 나무 인쇄 II


두 갈래 나무를 드리겠습니다. 층순으로 훑어보는 노드 값을 되돌려 주십시오.(즉, 왼쪽에서 오른쪽으로 모든 노드에 층층이 접근한다.)
예: 두 갈래 나무:[3,9,20,null,null,15,7],
    3
   / \
  9  20
    /  \
   15   7
 :

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

[문제 해결 방법 1] BFS - 대기열


List ans = new ArrayList<>(); Queue queue = new LinkedList<>(); List list = new ArrayList<>();
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> ans = new ArrayList<>();
        if(root == null){
            return ans;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            int size = queue.size();
            List<Integer> list = new ArrayList<>();
            while(size > 0){
                TreeNode cur = queue.poll();
                list.add(cur.val);
                size--;
                if(cur.left != null){
                    queue.offer(cur.left);
                }
                if(cur.right != null){
                    queue.offer(cur.right);
                }
            }
            ans.add(list);
        }
        return ans;
    }
}

[문제 해결 사고방식 2] DFS 귀속


같은 층에 같은 수조를 넣다
class Solution {
    private int num = 0;// 
    private List<List<Integer>> results = new LinkedList<List<Integer>>();
    public List<List<Integer>> levelOrder(TreeNode root) {
        //null 
        if(root == null) {
            return results;
        }
        // 
        DFS(root);
		return results;
    }
   void DFS(TreeNode node) {
       // result 
       if(results.size() <= num) {
           results.add(new LinkedList<Integer>());
       }
        //add result
       results.get(num).add(node.val);
        // 
       if(node.left != null) {
           //num+1 
           num++;
           DFS(node.left);
            // ,  
           num--;
       }
        // ,  
       if(node.right != null) {
           num++;
           DFS(node.right);
           num--;
       }
       return;

   }
}

좋은 웹페이지 즐겨찾기