题解 | #二叉树中和为某一值的路径(二)#

二叉树中和为某一值的路径(二)

http://www.nowcoder.com/practice/b736e784e3e34731af99065031301bca

深搜

思路: (1)从根节点开始往下搜索,遇到路径和等于expectNumber,并且当前节点为叶子节点时加入结果集; (2)注意,节点的值可以是负数,所以不能中断后续搜索。

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
class Solution {
private:
    vector<vector<int>> res;
    vector<int> vec;
    void helper(TreeNode* root, const int & expectNumber, int s){
        if(root == NULL) return;
        
        s += root -> val;
        vec.push_back(root -> val);
        
        if(s == expectNumber){
            if(root -> left == NULL && root -> right == NULL){
                res.push_back(vec);
            }
        }
        
        helper(root -> left, expectNumber, s);
        helper(root -> right, expectNumber, s);
        
        vec.pop_back();
    }
public:
    vector<vector<int>> FindPath(TreeNode* root,int expectNumber) {
        helper(root, expectNumber, 0);
        return res;
    }
};
全部评论

相关推荐

牛客316659795号:不是,证明hr初筛已经过了,要投给部门筛一遍
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务