leetcode_question_112 Path Sum

 

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 and  sum = 22 ,

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

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

DFS:

 

bool findPathSum(TreeNode* root, vector<int> &path, int sum)
    {
    	path.push_back(root->val);
		if(root->left == NULL && root->right == NULL){
			vector<int>::iterator it = path.begin();
			int tmpsum = 0;
			for(; it != path.end(); ++it)
				tmpsum += *it;
			path.pop_back();
			if(tmpsum == sum)
				return true;
			else 
				return false;
		}
		bool flag = false;
		if(root->left)
			flag = findPathSum(root->left,path,sum);
		if(!flag && root->right)
			flag = findPathSum(root->right,path,sum);
		path.pop_back();
		return flag;
	}
	bool hasPathSum(TreeNode *root, int sum) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        if(root == NULL)
			return false;
		vector<int> path;
		return findPathSum(root, path,sum);
    }