码迷,mamicode.com
首页 > 其他好文 > 详细

1530. 好叶子节点对的数量

时间:2020-07-27 13:50:30      阅读:114      评论:0      收藏:0      [点我收藏+]

标签:return   span   com   ima   节点   alt   res   air   null   

技术图片

 

class Solution {
    public int countPairs(TreeNode root, int distance) {
        dfs(root,0,distance);
        return res;
    }
    private int res = 0;
    public List<Integer> dfs(TreeNode root,int level, int distance) {
        List<Integer> list = new ArrayList<>();
        if(root == null) return list;
        if(root.left == null && root.right == null) list.add(level);

        List<Integer> left = new ArrayList<>(dfs(root.left,level+1,distance));
        List<Integer> right = new ArrayList<>(dfs(root.right,level+1,distance));
        
        for(int l : left) {
            for(int r : right) {
                if(l - level + r - level <= distance) res++;
            }
        }
        list.addAll(left);
        list.addAll(right);
        return list;
    }
}

 

1530. 好叶子节点对的数量

标签:return   span   com   ima   节点   alt   res   air   null   

原文地址:https://www.cnblogs.com/yonezu/p/13384269.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!