Leetcode刷题笔记 113. 路径总和 II

知识点：二叉树、回溯
时间：2020年9月26日
题目链接：https://leetcode-cn.com/problems/path-sum-ii/

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

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

示例1

输入：

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

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

输出：

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

解法

1. 在二叉树中的回溯，满足条件的放入答案数组
2. 回溯退出的条件：节点为空(比较好实现)或者为叶子节点 (比较好理解)
3. 如何回溯：放入此节点的值，判断是否满足条件，进入左右节点

代码

#include <stdio.h>
#include <iostream>
#include <vector>
#include <unordered_map>
using namespace std;
struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
/*
 退出条件是根节点
class Solution {
public:
    vector<int> tmp;
    vector<vector<int>> ans;
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        if(root==nullptr){return ans;}
        tmp.push_back(root->val);
        dfs(root, sum);
        return  ans;
    }
    void dfs(TreeNode* root,int sum){
        if(root->left==nullptr&&root->right==nullptr){
            int tsum = 0;
            for(int i=0;i<tmp.size();i++)
                tsum+=tmp[i];
            if(sum == tsum){
                ans.push_back(tmp);
            }
            return ;
        }
        if(root->left){
            tmp.push_back(root->left->val);
            dfs(root->left,sum);
            tmp.pop_back();
        }
        if(root->right){
            tmp.push_back(root->right->val);
            dfs(root->right,sum);
            tmp.pop_back();
        }
        return;
    }
};*/
class Solution {
public:
    vector<int> tmp;
    vector<vector<int>> ans;
    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        dfs(root, sum);
        return  ans;
    }
    //退出条件是空节点
    void dfs(TreeNode* root,int sum){
        if(root == nullptr) return;
        tmp.push_back(root->val);
        sum -= root->val;
        if(root->left==nullptr && root->right==nullptr && sum==0)
            ans.push_back(tmp);
        dfs(root->left,sum);
        dfs(root->right,sum);
        tmp.pop_back();
        return;
    }
};
int main()
{
    TreeNode node1(5);TreeNode node2(4);TreeNode node3(8);
    TreeNode node4(11);TreeNode node5(13);TreeNode node6(4);
    TreeNode node7(7);TreeNode node8(2);TreeNode node9(5);TreeNode node10(1);
    node1.left=&node2;node1.right=&node3;
    node2.left=&node4;node3.left=&node5;node3.right=&node6;
    node4.left=&node7;node4.right=&node8;node6.left=&node9;node6.right=&node10;
    Solution s;
    vector<vector<int>> ans = s.pathSum(&node1, 22);
    for(auto x:ans){
        for(auto y:x)
            cout<<y<<" ";
        cout<<endl;
    }
    return 0;
}

今天也是爱zz的一天哦！