题目描述:
给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
示例:
思路及实现:
根据题目很容易想到dfs算法,若dfs有一条路径返回true,则整体返回true。
转化为代码即为:
bool left,right;
left = dfs(node->left,sum,temp);
right = dfs(node->right,sum,temp);
return left|right;
其中,sum为最终的求和参数,temp为到目前为止累加的和,那么接下来需要判断,该路径是否已完成。由二叉树性质可知,上述算法无重复路径,那么判断路径是否完成,只要判断当前节点是否为叶子节点即可。
if(node->left==nullptr&&node->right==nullptr&&sum==temp)return true;
else if(node->left==nullptr&&node->right==nullptr&&sum!=temp)return false;
整体代码及其评估如下:
/**
* 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 hasPathSum(TreeNode* root, int sum) {
return dfs(root,sum,0);
}
bool dfs(TreeNode* root,int sum,int temp)
{
if(!root)return false;
int num = root->val;
temp = temp+num;
if(!(root->left)&&!(root->right)&&temp==sum)
{
return true;
}
else if(!(root->left)&&!(root->right)&&temp!=sum)
return false;
bool left = dfs(root->left,sum,temp);
bool right = dfs(root->right,sum,temp);
return left|right;
}
};