LintCode - 두 갈래 트리 레이아웃 II (중간)

1751 단어
판권 성명: 본고는 블로거의 오리지널 문장으로 블로거의 허락 없이 전재할 수 없습니다.
난이도: 중간
요구 사항:
두 갈래 나무를 제시하고 그 노드 값이 밑에서 위로 올라가는 차원으로 되돌아간다. (잎 노드가 있는 층에서 뿌리 노드가 있는 층으로 옮겨다닌 다음에 한 층씩 왼쪽에서 오른쪽으로 옮겨간다)
예:
예:
  {3,9,20,#,#,15,7},
    3
   / \
  9  20
    /  \
   15   7

 :
[
  [15,7],
  [9,20],
  [3]
]

구현:
/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */

public class Solution {
    /*
     * @param root: A tree
     * @return: buttom-up level order a list of lists of integer
     */
    public List> levelOrderBottom(TreeNode root) {
        List> ret = new ArrayList>();
        if (root == null) {
            return ret;
        }

        List cell = new ArrayList();
        TreeNode last = root;
        TreeNode nLast = root;

        LinkedList queue = new LinkedList();
        queue.add(root);
        while (!queue.isEmpty()) {
            TreeNode node = queue.pop();
            cell.add(node.val);

            if (node.left != null) {
                queue.add(node.left);
                nLast = node.left;
            }

            if (node.right != null) {
                queue.add(node.right);
                nLast = node.right;
            }

            // 
            if (node == last) {
                last = nLast;
                ret.add(0, cell);
                cell = new ArrayList();
            }
        }
        return ret;
    }
}

좋은 웹페이지 즐겨찾기