[LeetCode]113. Path Sum II(列出二叉树根到叶路径和等于sum的所有路径)

113. Path Sum II

原题链接

相似题目题解：112. Path Sum && 437. Path Sum III

Given a binary tree and a sum, find all root-to-leaf paths where each path’s sum equals the given sum.

For example:

Given the below binary tree and sum = 22,

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

return

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

题目大意：

给一个二叉树和一个数sum, 列出二叉树中从根节点到叶子节点的路径和等于sum的所有路径

思路：

运用深度优先遍历，结合112. Path Sum的题目，用二维数组paths存储所有符合的路径，用temp存储每次遍历的路径，符合条件就存进paths中

代码1：（16ms）

#include <iostream>
#include <vector>
#include <queue>
using namespace std;
//Definition for a binary tree node.
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};

class Solution {
public:
vector<vector<int>> pathSum(TreeNode* root, int sum) {//16ms
vector<vector<int>> paths; //存储所有满足条件的路径
vector<int> temp;//存储当前遍历路径
help(root, paths, temp, sum);
return paths;
}
void help(TreeNode* &root, vector<vector<int>> &paths, vector<int> temp,  int sum) {
if(root == nullptr)
return ;
temp.push_back(root->val);
if(root->left==nullptr && root->right==nullptr && sum==root->val)
paths.push_back(temp);
if(root->left != nullptr)
help(root->left, paths, temp, sum-root->val);
if(root->right != nullptr)
help(root->right, paths, temp, sum-root->val);
}
};
// 创建二叉树
TreeNode* CreateTreeByLevel(vector<int> num){
int len = num.size();
if(len == 0){
return NULL;
}//if
queue<TreeNode*> queue;
int index = 0;
// 创建根节点
TreeNode *root = new TreeNode(num[index++]);
// 入队列
queue.push(root);
TreeNode *p = NULL;
while(!queue.empty() && index < len){
// 出队列
p = queue.front();
queue.pop();
// 左节点
if(index < len && num[index] != -1){
// 如果不空创建一个节点
TreeNode *leftNode = new TreeNode(num[index]);
p->left = leftNode;
queue.push(leftNode);
}
index++;
// 右节点
if(index < len && num[index] != -1){
// 如果不空创建一个节点
TreeNode *rightNode = new TreeNode(num[index]);
p->right = rightNode;
queue.push(rightNode);
}
index++;
}//while
return root;
}

int main()
{
Solution s;
// -1代表NULL
vector<int> num = {5,4,8,11,-1,13,4,7,2,-1,-1,5,1};
TreeNode* root = CreateTreeByLevel(num);
vector<vector<int> > paths = s.pathSum(root,22);
for(int i = 0;i < paths.size();i++){
for(int j = 0;j < paths[i].size();j++){
cout<<paths[i][j]<<" ";
}
cout<<endl;
}
return 0;
}

代码2：（9ms）

发现更好的解法，每次遍历完左树及时删除元素效率更高

需要注意end函数用法

end函数：

函数原型：

iterator end();

const_iterator end();

功能：

返回一个当前vector容器中末尾元素的迭代器。

vector<vector<int>> pathSum1(TreeNode* root, int sum) {//9ms
vector<vector<int>> paths; //存储所有满足条件的路径
vector<int> temp;//存储当前遍历路径
help(root, paths, temp, sum);
return paths;
}
void help1(TreeNode* &root, vector<vector<int>> &paths, vector<int> &temp,  int sum) {
if(root == nullptr)
return ;
temp.push_back(root->val);
if(root->left==nullptr && root->right==nullptr && sum==root->val)
paths.push_back(temp);
if(root->left != nullptr)
help1(root->left, paths, temp, sum-root->val);
if(root->right != nullptr)
help1(root->right, paths, temp, sum-root->val);
//注意temp.end()指向的是最后一个元素的下一个位置，所以访问最后一个元素的正确操作为：temp.end() -1;
//temp.erase()  删除某个元素
temp.erase(temp.end() - 1);
}