leetcode113.路徑總和II

給定一個二叉樹和一個目標和，找到所有從根節點到葉子節點路徑總和等於給定目標和的路徑。

說明: 葉子節點是指沒有子節點的節點。

示例:

給定如下二叉樹，以及目標和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:

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

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/path-sum-ii
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

1. 定義一個二維向量res，如果存在頭節點，將該頭節點放入一維向量，然後將一維向量，與當前節點的左右子節點放入遞歸循環，如果該一維向量中所有加起來等於目標值，則將這個一維向量放入res中
2. 定義一個函數pathSum()，傳入參數root,sum,path，傳進來的path爲空的一維向量，先將root放入path中，如果sum等於path中所有元素之和將path放入res中，並且root無子節點。否則遞歸root->left，root->right
3. 大致代碼如下：

   	void pathSum(TreeNode* root, int sum, vector<int> path, int pathSum) {// 多定義了一個pathSum來記錄path所有元素之和
       if(root != nullptr) {
           path.push_back(root->val);
           pathSum += root->val;
           if(sum == pathSum && root->right == nullptr && root->left == nullptr) {
               res.push_back(path);
           } else {
               pathSum(root->left, sum, path, pathSum);
               pathSum(root->right, sum, path, pathSum);
           }
       }
   }
   

最終測試通過代碼

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> res;
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        vector<int> path;
        pathSum(root, sum, path, 0);
        return res;
    }
    void pathSum(TreeNode* root, int sum, vector<int> &path, int path_sum) { // pathSum改爲path_sum
        if(root != nullptr) {
            path.push_back(root->val);
            path_sum += root->val;
            if(sum == path_sum && root->right == nullptr && root->left == nullptr) {
                res.push_back(path);
            } else {
                pathSum(root->left, sum, path, path_sum);
                pathSum(root->right, sum, path, path_sum);
            }
            path.pop_back(); // 每次都需要返回到父節點
        }
    }
};