LeetCode: [102. 두 갈래 나무의 층계가 두루 훑어보기]

2376 단어 LeetCode

제목


두 갈래 트리를 정해서 차원별로 훑어보는 노드 값을 되돌려줍니다.(즉, zhu'ceng'de, 왼쪽에서 오른쪽으로 접근).
예를 들어, 지정된 두 갈래 트리:[3,9,20,null,null,15,7] ,
    3
   / \
  9  20
    /  \
   15   7

다음 단계를 반복합니다.
[
  [3],
  [9,20],
  [15,7]
]

문제 풀이 사고방식


두 개의 열을 사용하여 현재 처리 레이어를 저장하고 다음 처리 레이어를 저장합니다.층마다 다 처리합니다. 

코드 구현


트리 결점 클래스
public class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;

    TreeNode(int x) {
        val = x;
    }
}

Java 알고리즘 구현:
import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;


public class Solution {
	
    public List> levelOrder(TreeNode root) {
    	List> result = new ArrayList<>();
    	
    	if (root == null) {
    		return result;
    	}
    	
    	Deque cur = new LinkedList<>();  // 
    	Deque sub = new LinkedList<>();  // 
    	Deque exc;
    	
    	TreeNode node;
    	cur.addLast(root);    //                    
    	
    	while(!cur.isEmpty()) {
    		List layout = new LinkedList<>();  // 
    		while(!cur.isEmpty()) {
    			node = cur.removeFirst();   //             
    			layout.add(node.val);       // 
    			
    			if (node.left != null) {    // 
    				sub.addLast(node.left);
    			}
    			
    			if (node.right != null) {   // 
    				sub.addLast(node.right);
    			}
    		}
    		
    		exc = cur;       //             
    		cur = sub;
    		sub = exc;
    		
    		result.add(layout);  // 
    	}
		return result;
    }
}

좋은 웹페이지 즐겨찾기