三刷366. Find Leaves of Binary Tree

LinkedIn tag Medium
这道题三刷了都还没自主做出来。问题在于不知道这个题跟节点高度的联系。
我们定义helper function为find the height of the root node. 这里的height规定为leaf node的高度为0， 越往root走高度越高。null node高度为-1. 这样的话，我们从root开始，找root节点的高度。为了找到root节点的高度，其实在dfs里我们会先一条路走到黑从leaf node开始找，并且更新res. 这个过程其实我们就把所有node都遍历过了，也找到了他们的height. 而我们的res的size()跟height有密切关系.仔细观察example cases里面的答案，height = 0的出现在res index = 0 的list里面， height = 1的出现在res index = 1的list里面....所以我们的答案其实就是把height = 0一直到height = root.height的节点放进list里面再按顺序放进list
of list里。所以每当我们得到一个高度的时候，我们都要检查这个高度对应的list有没有被加到res里，也就是看res.size() ==? height. 如果没有加，我们就新建；如果加了我们直接add就好。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List> findLeaves(TreeNode root) {
        List> res = new ArrayList<>();
        height(root, res);
        return res;
    }
    
    private int height(TreeNode root, List> res){
        if (root == null){
            return -1; //leaf node height = 0
        }    
        int left = height(root.left, res);
        int right = height(root.right, res);
        int height = Math.max(left, right) + 1;
        if (height == res.size()){
            List list = new ArrayList<>();
            res.add(list);
        }
        res.get(height).add(root.val);
        return height;
    }
}