【LeetCode】path-sum i&ii

题干

path-sum i

Given a binary tree and a sum, determine if the tree has a root-to-leaf path such that adding up all the values along the path equals the given sum.

For example:

Given the below binary tree andsum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1

return true, as there exist a root-to-leaf path5->4->11->2which sum is 22.

path-sum ii

Given a binary tree and a sum, find all root-to-leaf paths where each path’s sum equals the given sum.

For example:

Given the below binary tree andsum = 22,

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

return

[
   [5,4,11,2],
   [5,8,4,5]
]

问题一：树根结点到子结点的所有路径中和是否等于已给的sum。

问题二：在问题一的基础上，打印所有符合条件的路径值。

数据结构

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */

解题思路

问题一

深度遍历dfs即可解决。

问题二

在深度遍历dfs的基础上，用verctor来存储路径，再用一个vector 存储符合条件的路径。

参考代码

问题一：

class Solution {
public:
    bool hasPathSum(TreeNode *root, int sum) {
        if(root==NULL)
            return false;
        if (root->left==NULL&&root->right==NULL&&sum-root->val==0)
            return true;//符合条件
        return (hasPathSum(root->left, sum-root->val)||hasPathSum(root->right, sum-root->val));//左右子树遍历
    }
};

问题二：

class Solution {
public:
    vector<vector<int> > pathSum(TreeNode *root, int sum) {
        vector<vector<int>> vv;
        vector<int>v;
        dfs(root,sum,vv,v);
        return vv;
    }
    void dfs(TreeNode *root,int sum,vector<vector<int>>& vv,vector<int>v)
    {
        if (root==NULL)
            return;
        v.push_back(root->val);//记录路径值
        if (root->left==NULL&&root->right==NULL&&sum-root->val==0)
            vv.push_back(v);//记录符合条件的路径
        dfs(root->left, sum-root->val,vv,v);//左右子树遍历
        dfs(root->right, sum-root->val,vv,v);
    }
};

易错点

对于vv的调用要用引用&，因为v的路径存储是不同级的递归调用。但是vv存在同级调用，值会改变无法传递，所以要用原地址引用。