112.Path Sum

来源:互联网 发布:htc d820u数据连接 编辑:程序博客网 时间:2024/04/29 10:43

问题:搜索二叉树,判断是否存在某个从根节点到叶节点的路径,该路径所有节点值之和为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);
    }
};
0 0
原创粉丝点击