검지offer 제2판 - 32.2.두 갈래 트리를 위에서 아래로 인쇄합니다.

2271 단어
본 시리즈 내비게이션: 검지offer(제2판)java 내비게이션 게시판 구현
면접 문제 32.2: 지점은 위에서 아래로 두 갈래 나무를 인쇄한다
제목 요구: 위에서 아래로 층별로 두 갈래 트리를 인쇄하고, 같은 층의 노드는 왼쪽에서 오른쪽으로 순서대로 인쇄하며, 층마다 한 줄을 인쇄한다.
문제 풀이 사고방식: 같은 층차가 두루 다니는데 이전 문제와 달리 여기는 층마다 노드의 개수를 기록하고 층마다 인쇄가 끝난 후에 리턴 문자를 하나 더 인쇄해야 한다.
package structure;
import java.util.LinkedList;
import java.util.Queue;
/**
 * Created by ryder on 2017/6/12.
 *  
 */
public class TreeNode {
    public T val;
    public TreeNode left;
    public TreeNode right;
    public TreeNode(T val){
        this.val = val;
        this.left = null;
        this.right = null;
    }
}
package chapter4;
import structure.TreeNode;
import java.util.LinkedList;
import java.util.Queue;
/**
 * Created by ryder on 2017/7/18.
 *  
 */
public class P174_printTreeInLine {
    public static void printTreeInLine(TreeNode root){
        if(root==null)
            return;
        Queue> queue = new LinkedList<>();
        queue.offer(root);
        TreeNode temp;
        while (!queue.isEmpty()){
            for(int size=queue.size();size>0;size--){
                temp = queue.poll();
                System.out.print(temp.val);
                System.out.print("\t");
                if(temp.left!=null)
                    queue.offer(temp.left);
                if(temp.right!=null)
                    queue.offer(temp.right);
            }
            System.out.println();
        }
    }
    public static void main(String[] args){
        //            1
        //          /   \
        //         2     3
        //       /  \   / \
        //      4    5 6   7
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.right = new TreeNode(3);
        root.left.left = new TreeNode(4);
        root.left.right = new TreeNode(5);
        root.right.left = new TreeNode(6);
        root.right.right = new TreeNode(7);
        printTreeInLine(root);
    }
}

실행 결과
1   
2   3   
4   5   6   7

좋은 웹페이지 즐겨찾기