代码改变世界

Path Sum

2015-04-10 09:52  笨笨的老兔子  阅读(119)  评论(0)  编辑  收藏  举报

给定一个二叉树和一个整数，求是否存在一条从根节点到叶节点的路径，路径上所有点的和等于给定的整数

思路：用广度优先搜索，将当前节点的值传到下一层节点即可。即将每一个节点的值更新为从根节点到该点的值的和

  1. class Solution {
  2. public:
  3.     bool hasPathSum(TreeNode *root, int sum) {
  4.         if (!root)
  5.             return false;
  6.         queue<TreeNode*> qNode;
  7.         qNode.push(root);
  8.         TreeNode* tmpNode = NULL;
  9.         while (!qNode.empty())
  10.         {
  12.             int qSize = qNode.size();
  13.             for (size_t i = 0; i < qSize; i++)
  14.             {
  15.                 tmpNode = qNode.front();
  16.                 if (tmpNode->val == sum && !tmpNode->left && !tmpNode->right)
  17.                 {
  18.                     return true;
  19.                 }
  20.                 else
  21.                 {
  22.                     if (tmpNode->left)
  23.                     {
  24.                         tmpNode->left->val += tmpNode->val;
  25.                         qNode.push(tmpNode->left);
  26.                     }
  27.                     if (tmpNode->right)
  28.                     {
  29.                         tmpNode->right->val += tmpNode->val;
  30.                         qNode.push(tmpNode->right);
  31.                     }
  32.                 }
  33.                 qNode.pop();
  34.             }
  35.         }
  36.         return false;
  37.     }
  38. };