题解 | #二叉树中和为某一值的路径(三)#
二叉树中和为某一值的路径(三)
http://www.nowcoder.com/practice/965fef32cae14a17a8e86c76ffe3131f
解题思路
层序遍历+递归
借助队列,层序遍历将每个节点添加到队列中,然后依次从队列中取出节点作为子树的根节点,来遍历每颗子树
如果Path与sum相等,那么就可以++res来增加路径数
/**
* struct TreeNode {
* int val;
* struct TreeNode *left;
* struct TreeNode *right;
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* };
*/
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param root TreeNode类
* @param sum int整型
* @return int整型
*/
int FindPath(TreeNode* root, int sum) {
// write code here
if(!root)
return 0;
queue<TreeNode*> q;
//根节点入队
q.push(root);
int nCount = 0;//记录每层节点的个数
res = 0;
//path = 0;
while(!q.empty()){
//记录当前层数的节点
nCount = q.size();
while(nCount--){
TreeNode *node = q.front();
q.pop();
//该节点的左右孩子入队
if(node->left)
q.push(node->left);
if(node->right)
q.push(node->right);
dfs(node,sum,0);
}
}
return res;
}
void dfs(TreeNode* root,int sum,int path)
{
if(!root)
return;
path += root->val;
if(path == sum)
++res;
//左
dfs(root->left,sum,path);
//右
dfs(root->right,sum,path);
}
private:
int res;
//int path;
};