LeetCode_Path Sum II

题目

给定一个二叉树，给定一个值，找和的等于这个值的所有路径

思路

经典深度优先搜索题，这类题一般都需要一个cur_sum表示当前已经的加和，每递归一层 先加上根节点的值，如果当前节点是叶子节点检测下值是否为所需和，

递归结构，先递归左子树，将当前左子节点放入栈，递归完再回溯（删除栈顶），再以同样的方式递归右子树，这题没有1A,忘记考虑 空树的情况了

代码

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int> > pathSum(TreeNode *root, int sum) {
        int cur_sum=0;
        vector<vector<int>>  ans;
        vector<int> path;
        ans.clear();
        path.clear();
        if(root==NULL) return ans;
        path.push_back(root->val);
        checkAndRecord(root,sum,cur_sum,ans,path);
    }

   bool checkAndRecord(TreeNode *root, int sum ,int cur_sum,vector<vector<int>> &ans,vector<int> &path) //代码风格不好，建树的过程要写一写
       {
           cur_sum = cur_sum + root->val;
           if (root->left==NULL && root->right==NULL)//到达叶子节点
                {
                      if (  cur_sum==sum)
                      {
                          ans.push_back(path);
                          return true;
                      }
                        else
                          return false;
                }
            bool flag1=false,flag2=false;
            if (root->left!=NULL)  {
                    path.push_back(root->left->val);
                    flag1=checkAndRecord(root->left,sum,cur_sum,ans,path);
                    path.erase(path.end()-1);
            }
            if(root->right!=NULL){
                    path.push_back(root->right->val);
                    flag2 =checkAndRecord(root->right,sum,cur_sum,ans,path);
                    path.erase(path.end()-1);
            }
            return flag1 || flag2;
      }
};