题目要求查找是否存在根节点到子节点的路径和等于指定的值,若存在则将路径存入容器。
思路 :从根节点开始,将自身值存入临时容器,并开始检索其子节点,将临时容器传递下去;直到叶子节点,判断路径和是否等于指定值,若等于,则保存临时容器的值。
代码:
vector<vector<int>> path;
bool travel(TreeNode* node,int sum,vector<int> temp)
{
int val=node->val;
if(node->left==NULL && node->right==NULL)
{
if(val==sum)
{
temp.push_back(node->val);
path.push_back(temp);
}
}
//因为不是地址或指针传递,所以传入左右子节点的temp不再是同一个。
if(node->left!=NULL)
{
travel(node->left, sum-val,temp);
}
if(node->right!=NULL)
{
travel(node->right, sum-val,temp);
}
return 0;
}
vector<vector<int>> pathSum(TreeNode* root, int sum) {
if(root!=NULL)
{
vector<int> temp;
travel(root, sum,temp);
}
return path;
}