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<List<Integer>> findLeaves(TreeNode root) {
List<List<Integer>> res = new ArrayList<>();
height(root, res);
return res;
}
private int height(TreeNode root, List<List<Integer>> 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<Integer> list = new ArrayList<>();
res.add(list);
}
res.get(height).add(root.val);
return height;
}
}
