leetcode112 113（同剑指offer34）437路径总和I II III

最新推荐文章于 2023-05-25 18:19:09 发布

孟哲Hogan

最新推荐文章于 2023-05-25 18:19:09 发布

阅读量94

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/zhezhe_9860/article/details/118354737

版权

LeetCode 专栏收录该内容

54 篇文章 0 订阅

订阅专栏

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

叶子节点是指没有子节点的节点。
在这里插入图片描述
给定和路径的二叉树路径模板DFS：

vector<vector<int>>res;

void dfs(TreeNode* root, int sum, vector<int> path) {
	
	if(root == nullptr) return;
	sum = sum - root->val;
	path.push(root->val);
	if(!root->left && !root->right && sum ==0) {
		res.push_back(path);
		return;
	}
	dfs(root->left, sum, path);
	dfs(root->right, sum, path);
}

题解：

bool hasPathSum(TreeNode* root, int targetSum) {
   if(root == nullptr) return false;
   targetSum -= root->val;
   if(!root->left && !root->right && targetSum == 0) return true;
   return hasPathSum(root->left,targetSum) || hasPathSum(root->right,targetSum);
}

在这里插入图片描述

进一步，如何求出所有和等于target的路径和？
在这里插入图片描述

输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出：[[5,4,11,2],[5,8,4,5]]

class Solution {
vector<vector<int>> res;
private:
    void dfs(TreeNode* root, int targetSum, vector<int> path) {
        if(root == nullptr) return;
        targetSum -= root->val;
        path.push_back(root->val);
        if(!root->left && !root->right && targetSum == 0) {
            res.push_back(path);
            return;
        }
        dfs(root->left,targetSum,path);
        dfs(root->right, targetSum,path);
    }
public:
    vector<vector<int>> pathSum(TreeNode* root, int targetSum) {
        vector<int> path;
        dfs(root,targetSum,path);
        return res;
    }
};

再进一步，不限定从根节点开始，到叶子节点结束的情况

给定一个二叉树，它的每个结点都存放着一个整数值。

找出路径和等于给定数值的路径总数。

路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8

      10
     /  \
    5   -3
   / \    \
  3   2   11
 / \   \
3  -2   1

返回 3。和等于 8 的路径有:

1.  5 -> 3
2.  5 -> 2 -> 1
3.  -3 -> 11

class Solution {
int sum = 0;
private:
    void dfs(TreeNode* root, int targetSum) {
        if(root == nullptr) return;
        targetSum -= root->val;
        if(targetSum == 0) { //这里不要限制root->left root->right为NULL的情况
            ++sum;           //因为不要求到叶子节点结束
        }
        dfs(root->left,targetSum);
        dfs(root->right, targetSum);
    }
    void preOrder(TreeNode* root, int targetSum) {
        if(root == nullptr) return;
        dfs(root,targetSum);
        preOrder(root->left, targetSum);
        preOrder(root->right, targetSum);
    }
public:
    int pathSum(TreeNode* root, int targetSum) {
        preOrder(root,targetSum);
        return sum;
    }
};