LeetCode: Path Sum II

AIvin24

于 2014-07-30 16:41:11 发布

阅读量681

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/AIvin24/article/details/38303033

版权

LeetCode 专栏收录该内容

144 篇文章 0 订阅

订阅专栏

思路：和前面一题一样，不过这次需要记录具体路径，在前面中，是通过空节点来进行递归终止的，这样会形成一个解路径重复记录，如一个叶子节点，如果满足条件，则会记录两次，因为左右节点都为空，搜索左节点会记录一次解，搜索右节点也会记录一次，所以，这里用叶子节点进行递归终止条件，其他思路和前面一题一样。

code:

class Solution {
public:
    bool dfs(TreeNode *root,int curSum,int sum,vector<int> curPath, vector<vector<int> > &ret){
        if(root->left == NULL && root->right == NULL){
            if(curSum + root->val == sum){
                curPath.push_back(root->val);
                ret.push_back(curPath);
                return true;
            }
            return false;
        }
        vector<int> t1(curPath);
        t1.push_back(root->val);
        bool left = false, right = false;
        if(root->left != NULL)
            left = dfs(root->left,curSum+root->val,sum,t1,ret);
        if(root->right != NULL)
            right = dfs(root->right,curSum+root->val,sum,t1,ret);
        return left || right;
        
    }
    vector<vector<int> > pathSum(TreeNode *root, int sum) {
        vector<vector<int> > ret;
        vector<int> curRet;
        if(root == NULL)
            return ret;
        dfs(root,0,sum,curRet,ret);
        return ret;
    }
};