问题:搜索二叉树,判断是否存在某个从根节点到叶节点的路径,该路径所有节点值之和为sum。
思路:思路很简单:深度优先搜索(其实就是回溯法),当搜索到叶节点时做一次判断;如果某次判断为真,结束程序,返回真;除非搜索完了都没有为真,返回假。
/**
* 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:
bool check(TreeNode* root, int sum,int path_sum){
if(root->left==NULL && root->right==NULL){
if(path_sum==sum){
return true;
}else{
return false;
}
}else{
if(root->left!=NULL){
path_sum+=root->left->val;
bool ret=check(root->left,sum,path_sum);
if(ret) return true;
path_sum-=root->left->val;
}
if(root->right!=NULL){
path_sum+=root->right->val;
bool ret=check(root->right,sum,path_sum);
if(ret) return true;
path_sum-=root->right->val;
}
return false;
}
}
bool hasPathSum(TreeNode* root, int sum) {
if(root==NULL) return false;
int path_sum=root->val;
return check(root,sum,path_sum);
}
};