112. 路径总和

最新推荐文章于 2022-07-25 20:03:55 发布

z-k

最新推荐文章于 2022-07-25 20:03:55 发布

阅读量93

点赞数

分类专栏： Leetcode Leetcode带思路

本文链接：https://blog.csdn.net/weixin_40673608/article/details/86501503

版权

Leetcode 同时被 2 个专栏收录

131 篇文章 3 订阅

订阅专栏

Leetcode带思路

98 篇文章 2 订阅

订阅专栏

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

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

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

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

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

思路：

用一个数组存每个结点上可能出现的和结果

root 数组 = 左孩子每个结果+val 和右孩子每个结果 + val

最后遍历数组，看是否有一个数和sum相等，若有，返回true，若都没有，返回false

code:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
static auto x = []() {
    std::ios::sync_with_stdio(false);
    std::cin.tie(nullptr);
    return 0;
}();
class Solution {
public:
    vector<int> path_sum(TreeNode *root)
    {
        vector<int> ret;
        if(root == NULL)
            return ret;
        vector<int> t1 = path_sum(root -> left);
        vector<int> t2 = path_sum(root -> right);
        for(int i = 0; i < t1.size(); i++)
            ret.push_back(t1[i] + root -> val);
        for(int i = 0; i < t2.size(); i++)
            ret.push_back(t2[i] + root -> val);
        if(ret.empty())
            ret.push_back(root -> val);
        return ret;
    }
    bool hasPathSum(TreeNode* root, int sum) {
        if(root == NULL)
            return false;
        vector<int> t = path_sum(root);
        for(int i = 0; i < t.size(); i++)
        {
            if(t[i] == sum)
                return true;
        }
        return false;
    }
};

效率：

100%