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

Posted 2021-10-26 vector6_

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

输入一颗二叉树的根节点和一个整数，按字典序打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

Input: {10,5,12,4,7},22

Output: [[10,5,7],[10,12]]

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(NULL), right(NULL) {
	}
};*/
//dfs 每次判断和是否达到expectNumber，且当前节点为叶子节点
class Solution {
public:
    vector<vector<int>> res;
    vector<int> curRes;
    
    vector<vector<int> > FindPath(TreeNode* root,int expectNumber) {
        if(root == nullptr)
            return res;
        curRes.push_back(root->val);
        expectNumber-=root->val;
        if(expectNumber==0 && root->left==nullptr && root->right==nullptr) //判断和是否达到expectNumber，且当前节点为叶子节点
        {
            res.push_back(curRes);
        }
        auto result1 = FindPath(root->left, expectNumber);
        auto result2 = FindPath(root->right, expectNumber);
        curRes.pop_back();
        return res;
    }
};