1.路径总和
给你二叉树的根节点 root 和一个表示目标和的整数 targetSum ,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和 targetSum。叶子节点是指没有子节点的节点。
考察知识点:DFS
class Solution {
public:
bool hasPathSum(TreeNode* root, int targetSum) {
if(!root) return false;
if(!root->left&&!root->right) return targetSum==root->val;
targetSum-=root->val;
return hasPathSum(root->left,targetSum)||hasPathSum(root->right,targetSum);
}
};
2.路径总和 II
给你二叉树的根节点 root 和一个整数目标和 targetSum,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。叶子节点 是指没有子节点的节点。
考察知识点:DFS
class Solution {
public:
vector<vector<int>> v;
vector<int> tmp;
vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
if(!root) return{};
tmp.emplace_back(root->val); //比push_back效率高
if((!root->left&&!root->right)&& targetSum==root->val) v.emplace_back(tmp);
targetSum-=root->val;
pathSum(root->left,targetSum);
pathSum(root->right,targetSum);
tmp.pop_back();
return v;
}
};
3.路径总和 III
给定一个二叉树,它的每个结点都存放着一个整数值。找出路径和等于给定数值的路径总数。
路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
考察知识点:DFS
双重递归解法:从每个节点开始进行类似的计算,所以第一个递归用来遍历这些节点,这二个递归用来处理这些节点,进行深度优先搜索。
class Solution {
public:
int count = 0;
int pathSum(TreeNode* root, int sum) {
if (!root) return 0;
dfs(root, sum) ;
pathSum(root->left, sum) ;
pathSum(root->right, sum);
return count;
}
void dfs(TreeNode* root, int sum) {
if (!root) return;
if (sum - root->val == 0) count++;
dfs(root->left, sum - root->val);
dfs(root->right, sum - root->val);
}
};
前缀和解法:遍历到某个节点时,在hash中查找前缀和为prefix_sum-sum的节点的个数,结果等于这个个数加上左右子树中满足条件的节点个数。
class Solution {
public:
unordered_map<int, int> count;
int pathSum(TreeNode* root, int sum) {
count[0] = 1;
return helper(root, sum, 0);
}
int helper(TreeNode* root, int sum, int prefix_sum) {
if (!root) return 0;
int res = 0;
prefix_sum += root->val;
res += count[prefix_sum - sum];
count[prefix_sum]++;
res += helper(root->left, sum, prefix_sum) + helper(root->right, sum, prefix_sum);
count[prefix_sum]--;
return res;
}
};