검지 Offer55 - 두 갈래 나무를 여러 줄로 인쇄합니다.

7822 단어 검지 Offer
위에서 아래로 층별로 두 갈래 트리를 인쇄하고 같은 층의 결점은 왼쪽에서 오른쪽으로 출력합니다.층마다 줄을 출력합니다.
/*
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;
 
    public TreeNode(int val) {
        this.val = val;
 
    }
 
}
*/
import java.util.ArrayList;
import java.util.Queue;
import java.util.LinkedList;
public class Solution {
    ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
        ArrayList<ArrayList<Integer>> re=new ArrayList<ArrayList<Integer>>();
        ArrayList<Integer> temp=new ArrayList<>();
        Queue<TreeNode> q=new LinkedList<TreeNode>();
        if(pRoot==null)
            return re;
        int next=0,now=1;//next:    now: 
        q.add(pRoot);
        while(!q.isEmpty()){
            TreeNode t=q.remove();
            now--;
            temp.add(t.val);
            if(t.left!=null){
                q.add(t.left);
                next++;
            }
            if(t.right!=null){
                q.add(t.right);
                next++;
            }
            if(now==0){
                now=next;
                next=0;
                re.add(new ArrayList<Integer>(temp));
                temp.clear();
            }
        }
        return re;
    }
     
}

좋은 웹페이지 즐겨찾기