LintCode 71-이 진 트 리 의 톱니 모양 층 이 널리 퍼 져 있다.

자신 이 ac 를 한 번 한 후에 구 글 순위 가 상위 권 에 있 는 자바 버 전의 실현 이 적 다 는 것 을 알 게 되 었 습 니 다.여기에 제 실현 을 붙 였 습 니 다.
사고방식 은 매우 간단 하 다.양 끝 대기 열 을 차원 으로 옮 겨 다 니 는 지원 구조 로 사용 한 다음 에 하나의 불 형 변수 로 다음 층 을 옮 겨 다 니 는 것 이 팀 의 머리 인지 팀 의 끝 인지 제어 한다.
public class Solution {
    /**
     * @param root: The root of binary tree.
     * @return: A list of lists of integer include 
     *          the zigzag level order traversal of its nodes' values 
     */
    public ArrayList> zigzagLevelOrder(TreeNode root) {
        // write your code here
        ArrayList> res = new ArrayList>();
        if (root == null) {
            return res;
        }

        ArrayList floor = new ArrayList<>();
        Deque deque = new ArrayDeque<>();
        TreeNode last = root;
        deque.offer(root);
        boolean nextPoll = true;

        while (!deque.isEmpty()) {
            TreeNode temp = nextPoll? deque.poll(): deque.pollLast();
            floor.add(temp.val);

            if (nextPoll) {
                if (temp.left != null) {
                    deque.offer(temp.left);
                }
                if (temp.right != null) {
                    deque.offer(temp.right);
                }
            } else {
                if (temp.right != null) {
                    deque.offerFirst(temp.right);
                }
                if (temp.left != null) {
                    deque.offerFirst(temp.left);
                }
            }

            if (last == temp) {
                res.add(floor);
                floor = new ArrayList<>();
                last = nextPoll? deque.peek(): deque.peekLast();
                nextPoll = !nextPoll;
            }
        }

        return res;
    }
}

좋은 웹페이지 즐겨찾기