검지offer--두 갈래 나무를 여러 줄로 인쇄하기

1289 단어

제목 설명


위에서 아래로 층별로 두 갈래 트리를 인쇄하고 같은 층의 결점은 왼쪽에서 오른쪽으로 출력합니다.층마다 줄을 출력합니다.
import java.util.ArrayList;


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

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

    }

}
*/
public class Solution {
	ArrayList<ArrayList<Integer> > Print(TreeNode pRoot) {
		ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
		if(pRoot==null) return result;
		ArrayList<TreeNode> queue = new ArrayList<TreeNode>();
		ArrayList<Integer> temp = new ArrayList<Integer>();
		ArrayList<Integer> start = new ArrayList<Integer>();
		start.add(pRoot.val);
		result.add(start);
		int low = 0;
		int high = 1;
		int end = high;	
		queue.add(pRoot);
		while(low<high){			
			TreeNode t = queue.get(low);
			if(t.left!=null){
				queue.add(t.left);
				temp.add(t.left.val);
				high++;
			}
			if(t.right!=null){
				queue.add(t.right);
				temp.add(t.right.val);
				high++;
			}
			low++;
			if(low==end){
				end = high;
				if(temp.size()!=0)
					result.add(temp);
				temp = new ArrayList<Integer>();
			}
		}
		return result;
	}
	
}

좋은 웹페이지 즐겨찾기