437. 路径总和 III

给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。

路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

示例 1：

输入：root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出：3
解释：和等于 8 的路径有 3 条，如图所示。

示例 2：

输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出：3

提示:

• 二叉树的节点个数的范围是 [0,1000]
• -109 <= Node.val <= 109 
• -1000 <= targetSum <= 1000 

 

 long long retInt = 0;

void backTrace(TreeNode* root, int targetSum, long long &sum)

{

  if (nullptr == root)

  {

    return;

  }

  sum += root->val;

  if (sum == targetSum)

  {

    retInt++;

  }

  backTrace(root->left, targetSum, sum);

  if (nullptr != root->left)

  {

    sum -= root->left->val;

  }

  backTrace(root->right, targetSum, sum);

  if (nullptr != root->right)

  {

    sum -= root->right->val;

  }

}

 

int pathSum(TreeNode* root, int targetSum)

{

  if (nullptr == root)

  {

    return 0;

  }

  queue que;

  que.push(root);

  while (!que.empty())

  {

    queue tmpQue;  

    while (!que.empty())

    {

       long long sum = 0;

      TreeNode* top = que.front();

      backTrace(top, targetSum, sum);

      if (top->left != nullptr)

      {

        tmpQue.push(top->left);

      }

      if (top->right != nullptr)

      {

        tmpQue.push(top->right);

      }

      que.pop();

    }

    que = tmpQue;

  }

  return retInt;

}