牛客网&剑指Offer&二叉树中和为某一值的路径

最新推荐文章于 2020-06-27 19:36:49 发布

不说就是不说

最新推荐文章于 2020-06-27 19:36:49 发布

阅读量101

点赞数

分类专栏： C++ 面试编程题

本文链接：https://blog.csdn.net/weixin_42060900/article/details/95515958

版权

C++ 同时被 2 个专栏收录

55 篇文章 1 订阅

订阅专栏

面试编程题

43 篇文章 0 订阅

订阅专栏

牛客网&剑指Offer&二叉树中和为某一值的路径

在这里插入图片描述
代码实现

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;
    void find(TreeNode* root, int sum)
    {
        if(root == nullptr)
            return;
        path.push_back(root->val);
        if(!root->left && !root->right && sum == root->val)
            res.push_back(path); //尾部插入
        else
        {
            if(root->left)
                find(root->left, sum - root->val);
            if(root->right)
                find(root->right, sum - root->val);
        }
        path.pop_back(); //尾部删除
    }
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        find(root,expectNumber);
        return res; 
    }
};

编程笔记

代码实现解题思路：使用递归思想。使用标准模板库中的vector实现了一个栈来保存路径，每次都使用push_back在路径的末尾添加节点，用pop_back在路径的末尾删除节点，这样就保证了栈的先入后出特性。这里没有直接使用STL中的stack原因是，在stack中只能得到栈顶元素，而打印路径的时候需要得到路径上的所有节点，因此在代码实现的时候std::stack不是最好的选择。
熟悉二维数组的表现形式。