LeetCode（113）：路径总和 II

Medium！

题目描述：

给定一个二叉树和一个目标和，找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

说明: 叶子节点是指没有子节点的节点。

示例:
给定如下二叉树，以及目标和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

[
   [5,4,11,2],
   [5,8,4,5]
]

解题思路：

这道二叉树路径之和在之前的基础上又需要找出路径，但是基本思想都一样，还是需要用深度优先搜索DFS，只不过数据结构相对复杂一点，需要用到二维的vector，而且每当DFS搜索到新节点时，都要保存该节点。

同时，每当找出一条路径之后，都将这个保存为一维vector的路径保存到最终结果二位vector中。

并且，每当DFS搜索到子节点，发现不是路径和时，返回上一个结点时，需要把该节点从一维vector中移除。

C++解法一：

 1 class Solution {
 2 public:
 3     vector<vector<int> > pathSum(TreeNode *root, int sum) {
 4         vector<vector<int>> res;
 5         vector<int> out;
 6         helper(root, sum, out, res);
 7         return res;
 8     }
 9     void helper(TreeNode* node, int sum, vector<int>& out, vector<vector<int>>& res) {
10         if (!node) return;
11         out.push_back(node->val);
12         if (sum == node->val && !node->left && !node->right) {
13             res.push_back(out);
14         }
15         helper(node->left, sum - node->val, out, res);
16         helper(node->right, sum - node->val, out, res);
17         out.pop_back();
18     }
19 };

下面这种方法是迭代的写法，用的是中序遍历的顺序，参考之前那道Binary Tree Inorder Traversal：http://www.cnblogs.com/grandyang/p/4297300.html，中序遍历本来是要用栈来辅助运算的，由于我们要取出路径上的节点值，所以我们用一个vector来代替stack，首先利用while循环找到最左子节点，在找的过程中，把路径中的节点值都加起来，这时候我们取出vector中的尾元素，如果其左右子节点都不存在且当前累加值正好等于sum了，我们将这条路径取出来存入结果res中，下面的部分是和一般的迭代中序写法有所不同的地方，因为如果当前最左节点已经是个叶节点了，我们要转移到其他的节点上时需要把当前的节点值减去，而如果当前最左节点不是叶节点，下面还有一个右子节点，这时候移动指针时就不能减去当前节点值，为了区分这两种情况，我们需要用一个额外指针pre来指向前一个节点，如果右子节点存在且不等于pre，我们直接将指针移到右子节点，反之我们更新pre为cur，cur重置为空，val减去当前节点，s删掉最后一个节点，参见代码如下。

C++解法二：

 1 class Solution {
 2 public:
 3     vector<vector<int> > pathSum(TreeNode *root, int sum) {
 4         vector<vector<int>> res;
 5         vector<TreeNode*> s;
 6         TreeNode *cur = root, *pre = NULL;
 7         int val = 0;
 8         while (cur || !s.empty()) {
 9             while (cur) {
10                 s.push_back(cur);
11                 val += cur->val;
12                 cur = cur->left;
13             }
14             cur = s.back(); 
15             if (!cur->left && !cur->right && val == sum) {
16                 vector<int> v;
17                 for (auto it : s) {
18                     v.push_back(it->val);
19                 }
20                 res.push_back(v);
21             }
22             if (cur->right && cur->right != pre) cur = cur->right;
23             else {
24                 pre = cur;
25                 val -= cur->val;
26                 s.pop_back();
27                 cur = NULL;
28             }
29         }
30         return res;
31     }
32 };

LeetCode（113）：路径总和 II

猜你喜欢