1. 题目描述
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.
2. 样例
3. 分析
这道题属于典型使用DFS的思路。题目要求很明确,给了一棵树以及给了一个整数sum,让我们判断在这棵树中能否找到一条从根节点到叶子节点的路径出来,使该路径中所有节点的数值之和等于sum。
思路呢:利用DFS的策略,从根节点开始向下进行遍历(其实就是树的先序遍历)。
在遍历过程中,如果遍历到叶子节点,此时的叶子节点对应的数值与此时的sum相同,则为true,否则该条路径为false。
如果没有遍历到叶子节点,就继续递归,其中sum的值需要减去当前节点的val值,代表该节点已经被访问过了。
4. 源码
/**
* 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) {
if (root == NULL) {
return false;
}
if (root->val == sum && root->left == NULL && root->right == NULL) {
return true;
}
else {
if (hasPathSum(root->left, sum-root->val) == true || hasPathSum(root->right, sum-root->val) == true) {
return true;
}
return false;
}
}
};
5. 心得
我觉得使用DFS、或者之前的分治法的时候,递归是一种最常见的手段,然而递归的重点在于如何设计减小问题的规模,即如何设计递归里面的具体结构,以及何时跳出递归的条件。
第一次我提交代码的时候出现了WA,原因是题目里面没有说清楚是必须根节点到叶子节点的路径,结果出现了偏差。
以下是错误的代码:
/**
* 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) {
if (root == NULL) {
return false;
}
if (root->val == sum) {
return true;
}
else if (root->val > sum){
return false;
}
else {
if (hasPathSum(root->left, sum-root->val) == true || hasPathSum(root->right, sum-root->val) == true) {
return true;
}
return false;
}
}
};