题目链接:https://leetcode-cn.com/problems/path-sum-iii/
题目描述
给定一个二叉树,它的每个结点都存放着一个整数值。
找出路径和等于给定数值的路径总数。
路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
二叉树不超过1000个节点,且节点数值范围是 [-1000000,1000000] 的整数。
示例:
root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8
10
/ \
5 -3
/ \ \
3 2 11
/ \ \
3 -2 1
返回 3。和等于 8 的路径有:
1. 5 -> 3
2. 5 -> 2 -> 1
3. -3 -> 11
思路
1 层次遍历+递归
层序遍历确定开始的节点,从该节点递归往下找符合的路径
思路:首先先序递归遍历每个节点,再以每个节点作为起始点递归寻找满足条件的路径。
时间复杂度: O ( n 2 ) O(n^2) O(n2)
class Solution {
int ret = 0;
public:
int pathSum(TreeNode* root, int sum) {
if(!root) return 0;
queue<TreeNode*> q;
q.push(root);
while (!q.empty()){
TreeNode *node = q.front();
q.pop();
hasPathSum(node,sum); // 以该节点为起点往下找
if(node->left)
q.push(node->left);
if(node->right)
q.push(node->right);
}
return ret;
}
private:
// 返回:从root开始和为sum的路径数
void hasPathSum(TreeNode* root, int sum) {
if(!root) return;
sum -= root->val;
if(sum == 0)
++ret;
hasPathSum(root->left,sum);
hasPathSum(root->right,sum);
}
};
2 双重递归
(1)递归先序遍历
(2)然后在先序遍历里以每个遍历的节点为起始节点向下递归得到路径和为sum的路径数
时间复杂度: O ( n 2 ) O(n^2) O(n2)
/*
* 双重递归
*/
class Solution {
int ret = 0;
public:
int pathSum(TreeNode* root, int sum) {
if(!root) return 0;
hasPathSum(root,sum);
pathSum(root->left,sum);
pathSum(root->right,sum);
return ret;
}
private:
// 返回:从root开始和为sum的路径数
void hasPathSum(TreeNode* root, int sum) {
if(!root) return;
sum -= root->val;
if(sum == 0)
++ret;
hasPathSum(root->left,sum);
hasPathSum(root->right,sum);
}
};
3 哈希表+一遍递归
设置变量currSum表示当前从根节点到该访问节点路径的累计和(前n项和
)。
对于树的话,采取DFS加回溯,每次访问到一个节点,把该节点加入到当前的currSum
中
然后判断是否存在一个之前的前n项和,其值等于currSum
与sum之差
如果有,就说明现在的前n项和,减去之前的前n项和,等于sum,那么也就是说,这两个点之间的路径和,就是sum
/*
* hashmap+一遍递归
* 类似于数组的前n项和的思路,比如sum[4] == sum[1],那么1到4之间的和肯定为0
* 时间复杂度:O(n)
*/
class SolutionII {
unordered_map<int,int> m; // <当前路径和, 数量>
int ret = 0;
public:
int pathSum(TreeNode* root, int sum) {
if(!root) return 0;
m[0] = 1; // 对应于某个节点值=sum的情况
hasPathSum(root,sum,0);
return ret;
}
private:
void hasPathSum(TreeNode* root, int sum, int currSum) {
if(!root) return;
currSum += root->val;
ret += m[currSum - sum]; // 如果存在m[currSum - sum],说明中间有一段和为sun
m[currSum] += 1; // 从根节点到当前节点的路径总和
hasPathSum(root->left, sum,currSum);
hasPathSum(root->right, sum,currSum);
m[currSum] -= 1; // 回溯,去掉该节点的路径和
}
};