[leetcode] 4. 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.

 首先这个树很奇怪。。。。我也不知道他是怎么插进去的值的,不像传统二叉树那样有大小判断插入,而是好像在。。。随机插入。。。当然这个我们不管,题目是要问找一条从根到叶的路径加下来sum能否等于给的sum。

 如果二叉树基本功熟练的话,可以直接看出来这就是一个深度优先的搜索,然后这个是前序遍历加个和就行了。解法如下:

/**
 * Definition for binary tree
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
void PreSum(TreeNode *temp, int sum, int tmp, bool &flag)
{
	if (temp != NULL)
	{
		if (temp->val + tmp == sum && temp->left == NULL && temp->right == NULL)
			flag = true;
		else
		{
			tmp += temp->val;
			PreSum(temp->left, sum, tmp, flag);
			PreSum(temp->right, sum, tmp, flag);
		}	
	}
	
}

    bool hasPathSum(TreeNode *root, int sum) {
        bool flag = false;
        PreSum(root, sum, 0, flag);
        return flag;
    }
};

 因为这个递归弹出实在没法跟要求一样,所以我加了个flag作为判断。

posted @ 2014-11-11 23:45  Tiny-Box  阅读(122)  评论(0编辑  收藏  举报