二叉树后序遍历保存从根到结点路径方法

框架如下:关键点在于stack.push_back();以及stack.pop_back()的时间点。
void function(TreeNode* root , vector<int>& stack)
{
    if(root != NULL)
    {
        stack.push_back(root->val);
        function(ret , root->left , stack);
        function(ret , root->right , stack);
        stack.pop_back();
    }
}

练习题目:(注意结合上面的框架)
(1)力扣1448:统计二叉树中好节点数目。
解答:

class Solution {
public:
    int goodNodes(TreeNode* root) {
        int ret = 0;
        vector<int> stack;
        function(ret , root , stack);
        return ret;
    }

    void function(int& ret , TreeNode* root , vector<int>& stack)
    {
        if(root != NULL)
        {
            stack.push_back(root->val);
            int pre = root->val;
            ret += 1;
            for(int i = 0 ; i < stack.size() - 1 ; i++)
            {
                if(stack[i] > pre)
                {
                    ret -= 1;
                    break;
                }
            }
            function(ret , root->left , stack);
            function(ret , root->right , stack);
            stack.pop_back();
        }
    }
};

(2)力扣113:路径总和Ⅱ。
解答:

class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
        vector<vector<int>> ret;
        int sum = 0;
        vector<int> track;
        function(targetSum , sum , root , track , ret);
        return ret;
    }

    void function(int& targetSum , int& sum , TreeNode* root , vector<int>& track , vector<vector<int>>& ret)
    {
        if(root != NULL)
        {
            track.push_back(root->val);
            sum += root->val;
            if(root->left == NULL && root->right == NULL)
            {
                if(sum == targetSum)
                    ret.push_back(track);
            }
            function(targetSum , sum , root->left , track , ret);
            function(targetSum , sum , root->right , track , ret);
            sum -= root->val;
            track.pop_back();
        }
    }
};

(3)力扣129:求根节点到叶节点数字之和:

class Solution {
public:
    int sumNumbers(TreeNode* root) {
        int ret = 0;
        vector<int> track;
        function(ret , root , track);
        return ret;
    }

    void function(int& ret  ,TreeNode* root , vector<int>& track)
    {
        if(root != NULL)
        {
            track.push_back(root->val);
            if(root->left == NULL && root->right == NULL)
            {
                int sum;
                for(int i = track.size() - 1 ; i >= 0 ; i--)
                {
                    ret += track[i] * pow(10 , track.size() - 1 - i); 
                }
            }
            function(ret , root->left , track);
            function(ret , root->right , track);
            track.pop_back();
        }
    }
};
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值