Leetcode 1530. Number of Good Leaf Nodes Pairs DFS做出来

  • 一般树的问题都是要递归去做的,而且 从头到尾都是在递归,我就不小心想成了一个半递归的问题,肯定解不出来:/
  • dfs 每个父结点负责统计以它为根节点情况下,左子树和右子树的叶子结点满足条件的配对数
class Solution {
public:
    int countPairs(TreeNode* root, int distance) {
        dfs(root, distance);
        return ans;
    }
private:
    int ans=0;
    vector<int> dfs(TreeNode* root, int d){
        if(!root) return {};
        if(!root->left&&!root->right) return {1};
        auto l = dfs(root->left, d);
        auto r = dfs(root->right, d);
        for(auto& a: l)
            for(auto& b: r){
                if(a+b<=d) ans++;
            }
        vector<int> ret;
        for(auto& a: l)
            if(a+1<=d) ret.push_back(a+1);
        for(auto& a: r)
            if(a+1<=d) ret.push_back(a+1);
        return ret;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值