好叶子节点对的数量（后序遍历DFS）

r囧r小猫 2023-10-06 14:02 25阅读 0赞

## 好叶子节点对的数量（后序遍历DFS） ##

#### 问题： ####

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70 1]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70 2]

#### 思路： ####

我们需要知道节点的叶子节点到节点的距离，从中选取满足条件的叶节点对

通过后序遍历DFS分别构造节点的叶子节点到节点的左孩子的距离以及节点的叶子节点到节点的右孩子距离，通过这两个距离构造节点的叶子节点到节点的距离

DFS的返回值为vector< int >，表示节点的叶节点到节点的距离，通过DFS，我们可以知道叶节点到节点的左孩子的距离以及到节点的右孩子的距离，这两个距离+1即为叶节点到节点的距离，若有满足条件的距离，将此距离加入到保存叶节点到节点的距离的vector中

再从保存叶节点到节点的左孩子距离的vector以及保存到右孩子距离的vector中各选一个距离，若距离之和满足条件，则满足条件的叶节点对数量+1

#### 代码： ####

/**
     * Definition for a binary tree node.
     * struct TreeNode {
     *     int val;
     *     TreeNode *left;
     *     TreeNode *right;
     *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
     *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
     *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
     * };
     */
    class Solution {
        
    public:
        vector<int> dfs( TreeNode* root,int distance,int& ans )
        {
        
            if( root==NULL )
                return {
        };
            else if( root->left==NULL && root->right==NULL )
                return {
        0};
            vector<int> root_leaf;
            vector<int> left_leaf;
            left_leaf=dfs( root->left,distance,ans );
            vector<int> right_leaf;
            right_leaf=dfs( root->right,distance,ans );
            for( auto x:left_leaf )
            {
        
                x++;
                if( x<=distance )
                   root_leaf.push_back( x );
            }
            for( auto y:right_leaf )
            {
        
                y++;
                if( y<=distance )
                    root_leaf.push_back( y );
            }
            for( auto xx:left_leaf )
                for( auto yy:right_leaf )
                {
        
                    if( xx+yy+2<=distance )
                        ans++;
                }
            return root_leaf;
        }
        int countPairs(TreeNode* root, int distance) {
        
            int ans=0;
            dfs( root,distance,ans );
            return ans;
        }
    };

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/2020080110114529.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20200801101200661.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw_size_16_color_FFFFFF_t_70 2]: https://img-blog.csdnimg.cn/20200801101217882.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzQ0NzA5OTkw,size_16,color_FFFFFF,t_70