Leetcode: 113. Path Sum II (week3 --- medium)

题目 

题解

总结

---

题目 

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

Note: A leaf is a node with no children.

题目给出了一个例子：

 所以也就是相当于寻找一条从根root开始的到叶节点的路径，这一条路径需要满足和为题目中所要求的数字。

其中叶节点的要求是：没有任何的子节点。这是其中一个比较坑的地方

题解

1. 因为这里要求的是从根节点出发的，需要将树遍历一次，根据树的遍历算法，有：前序遍历、中序遍历、后序遍历三种遍历方式。而这里要求的是从根节点开始，所以不难知道使用前序遍历的递归思想是比较容易实现的
2. 这里可以给solution定义一个私有变量，sum，也就是目的要求的总和。这样就能够减少函数的参数。
3. 以前序遍历作为主体，改造其为如下，其中root也就是当前遍历到的节点，而current则是用来记录当前路径的，result则是最后的返回值，current_sum则是用来记录当前的总和值。

4. void preorder(TreeNode* root, vector<int>& current,vector<vector<int>>& result, int current_sum);

   在sum与current_sum相等的时候，还需要判断最后的节点是不是叶节点，若是则是符合的路径，所以代码如下：

if(current_sum == sum && root->left == NULL && root->right == NULL){
      result.push_back(current);
}

所以总的代码如下：

class Solution {
public:
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        this->sum = sum;
        vector<int> current;
        vector<vector<int>> result;
        if(root == NULL){
            return result;
        }
        int current_sum = root->val;
        current.push_back(root->val);
        if(current_sum == sum && root->left == NULL && root->right == NULL){
            result.push_back(current);
        }
        preorder(root->left, current, result, current_sum);
        preorder(root->right, current, result, current_sum);
        current.pop_back();
        return result;
    }

    void preorder(TreeNode* root, vector<int>& current,vector<vector<int>>& result, int current_sum){
        if(root == NULL){
            return ;
        }
        current_sum+=root->val;

            current.push_back(root->val);
            if(current_sum == sum && root->left == NULL && root->right == NULL){
                result.push_back(current);
            }
            preorder(root->left, current, result, current_sum);
            preorder(root->right, current, result, current_sum);
            current.pop_back();
    }
private:
    int sum;
};

总结

觉得在学习DFS与BFS的时候实际上很多是跟树的遍历方式很相似，只不过树的遍历是一种特殊情况，不需要进行标记操作。所以这些知识是可以互通。