1530. 好叶子节点对的数量

最新推荐文章于 2022-08-14 16:58:31 发布

iamcrystal94

最新推荐文章于 2022-08-14 16:58:31 发布

阅读量157

点赞数

分类专栏： leetcode笔记本文章标签： leetcode 二叉树

本文链接：https://blog.csdn.net/iamcrystal94/article/details/112171616

版权

leetcode笔记本专栏收录该内容

1 篇文章 0 订阅

订阅专栏

给定二叉树的根节点和一个距离distance，找到所有叶节点间最短路径小于等于distance的叶子节点对，返回这样的好叶子节点对数量。解题方法涉及计算节点到叶子节点的距离。

摘要由CSDN通过智能技术生成

题目描述：

给你二叉树的根节点 root 和一个整数 distance 。

如果二叉树中两个叶节点之间的最短路径长度小于或者等于 distance ，那它们就可以构成一组好叶子节点对。

返回树中好叶子节点对的数量。

解题思路：

对于任何的节点node，分别求节点node的左右叶子节点到node的距离，再计算每个左叶子节点与右叶子节点的距离。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    int res_dfs = 0;
    public int countPairs(TreeNode root, int distance) {   
        if(root != null){
            countnum(root, distance);
            countPairs(root.left, distance);
            countPairs(root.right, distance); 
        }
        return res_dfs; 
    }

    public void countnum(TreeNode root,  int distance){
        int start = 0;
        //计算节点root的到左右叶子节点的距离   
        List<Integer> leftLen = new ArrayList<Integer>();
        List<Integer> rightLen = new ArrayList<Integer>();
        if(root != null){
            if(root.left != null){
                countLen(root.left, start + 1, distance, leftLen);
            }
            if(root.right != null){
                countLen(root.right, start + 1, distance, rightLen);
            }
        }
        for(int i=0; i<leftLen.size(); i++){
            for(int j=0; j<rightLen.size(); j++){
                int count = leftLen.get(i) + rightLen.get(j);
                if(count < distance || count == distance){
                    res_dfs = res_dfs + 1; 
                } 
            }
        }      
    }

    public void countLen(TreeNode root,int len ,int distance,List<Integer> leaflen){
        if(len < distance){
            if(root.left != null){
                countLen(root.left, len+1, distance, leaflen); 
            }
            if(root.right != null){
                countLen(root.right, len+1, distance, leaflen);
            }
            if(root.left == null & root.right == null){
                leaflen.add(len);
            }
        }   
    }
}