검지 Offer: 면접 문제 23 - 위에서 아래로 두 갈래 나무 인쇄(java 구현)

2256 단어 문제를 풀다.
문제 설명:
위에서 아래로 두 갈래 나무의 각 노드를 인쇄하고, 같은 층의 노드는 왼쪽에서 오른쪽으로 인쇄합니다.
생각:
차원별로 훑어보는 방법에 따라 대기열 보조를 사용하다.1. 루트 결점을 대기열에 추가합니다.2. 행렬을 순환하여 현재 요소를 인쇄하고 결점에 왼쪽 트리가 있으면 대기열에 추가하고 오른쪽 트리가 있으면 대기열에 추가합니다.3. 대기열이 비어 있을 때까지 모든 결점이 인쇄되었음을 나타냅니다.
코드:
import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;

public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}

public class Solution {
    public ArrayList PrintFromTopToBottom(TreeNode root) {
        ArrayList result = new ArrayList();
        if(root == null){
            return result;
        }

        Queue q = new LinkedList();

        q.add(root);        
        while(!q.isEmpty()){
            TreeNode tmp = q.remove();
            result.add(tmp.val);
            if(tmp.left != null){
                q.add(tmp.left);
            }

            if(tmp.right != null){
                q.add(tmp.right);

            }

        }

        return result;
    }
}

좋은 웹페이지 즐겨찾기