(刷题笔记) Leetcode 112. 路径总和

题目

给定一个二叉树和一个目标和，判断该树中是否存在根节点到叶子节点的路径，这条路径上所有节点值相加等于目标和。

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

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

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

返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/path-sum

解题思路

这道题我总觉得我写过，但是找了一下写过的题目，只有类似的，没有完全一样的。
本质上这道题是所有路径和的变体。
我果然更习惯写非递归一些，递归我还特意去看了一下我前面写的路径和才写出来。
非递归的思路就是维护一个栈，存储结点，以及到那个结点的路径和，然后在叶子结点处比对是否等于指定值。
递归的思路类似，用一个vector存储到叶子结点的路径和，然后递归完成后比对是否等于指定值。
我建了个qq群，有兴趣的可以加一下575474684，大家可以互相督促一起刷题。

代码(C++)

/**
 * 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:
    bool hasPathSum(TreeNode* root, int sum) {

        stack<pair<TreeNode*,int>> st;

        TreeNode* temp=root;
        int num=0;

        while(!st.empty()||temp){

            while(temp){
                
                num+=temp->val;
                st.push(make_pair(temp,num));
                temp=temp->left;
            }

            temp=st.top().first;
            num=st.top().second;
            st.pop();
            if(temp->left==NULL&&temp->right==NULL){
                if(sum==num){
                    return true;
                }
            }

            temp=temp->right;
        }

        return false;
    }
};

递归

/**
 * 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:

    void pathsum(TreeNode* root,int num,vector<int>& res){

        if(root==NULL) return;
        num+=root->val;
        if(root->left==NULL&&root->right==NULL){
            res.push_back(num);
        }

        pathsum(root->left,num,res);
        pathsum(root->right,num,res);

    }

    bool hasPathSum(TreeNode* root, int sum) {

        vector<int> res;
        int num=0;
        pathsum(root,num,res);

        for(int i=0;i<res.size();++i){
            if(res[i]==sum){
                return true;
            }
        }

        return false;
        
    }
};