OJ-leetcode-113. 路径总和 II（中等二叉树）

最新推荐文章于 2024-12-18 01:14:02 发布

lady_killer9

最新推荐文章于 2024-12-18 01:14:02 发布

阅读量678

点赞数

分类专栏： OJ 文章标签：二叉树算法数据结构

本文链接：https://blog.csdn.net/lady_killer9/article/details/108818309

版权

OJ 专栏收录该内容

23 篇文章

订阅专栏

题目

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

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

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

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

[
[5,4,11,2],
[5,8,4,5]
]
链接：https://leetcode-cn.com/problems/path-sum-ii

思路

自顶向下回溯（vector去除点）可剪枝递归

剪枝 sum<0

以上是我做题时记的简单思路，最后没有剪枝，提交代码前看了一下官方题解，发现可能有负数。

代码

class Solution {
	vector<vector<int>> path;
	vector<int> nodes;
public:
	void pathSumHelper(TreeNode* root, int sum) {
		if (root == NULL)return;
		sum -= root->val;
		nodes.emplace_back(root->val);
		if (root->left == NULL&&root->right == NULL && 0 == sum)
		{
			path.emplace_back(nodes);
			nodes.pop_back();
			return;
		}
		// 剪枝  不行，有负数
		//if (sum<0)return;
		pathSumHelper(root->left,sum);
		pathSumHelper(root->right, sum);
		// 回溯
		nodes.pop_back();
	}
	vector<vector<int>> pathSum(TreeNode* root, int sum) {
		pathSumHelper(root, sum);
		return path;
	}
};

结果

仅做演示步骤，并非真实递归栈，可使用栈模拟并输出。

优秀题解

递归中的题解没有找到太好的，看见了一个递归转循环的。

后序遍历求递归路径

他模拟了栈，递归转为了循环，写的很不错，有时间总结一下递归转循环的。代码写的很好，push_back改为emplace_back就更好了。

提升笔记

1.递归剪枝

2.递归修改循环

全部代码

/*
Project: 113.路径综合II
Date:    2020/09/26
Author:  Frank Yu
*/
/**
* Definition for a binary tree node.*/
#define TElemType int
#include<iostream>
#include<vector>
using namespace std;
struct TreeNode {
   int val;
   TreeNode *left;
   TreeNode *right;
   TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 };
//创建二叉树 规定数据域为-1，则为空 先序创建
int InitTree(TreeNode* &T)
{
	TElemType a;
	scanf("%d", &a);
	if (-1 == a) T = NULL;
	else {
		T = new TreeNode(a);
		//T->val = a;
		InitTree(T->left);
		InitTree(T->right);
	}
	return 0;
}
//这里就按照课本敲代码了，在对应功能实现函数进行逻辑判断
void PreOrder(TreeNode* T)
{
	if (T != NULL)
	{
		printf("%d ", T->val);
		PreOrder(T->left);//递归先序遍历左右子树
		PreOrder(T->right);
	}
}
//调用InitTree
void CreateBiTree(TreeNode* &T)
{
	printf("请按照先序遍历输入二叉树(-1无):");
	InitTree(T);
	printf("二叉树先序遍历序列:");
	PreOrder(T);
	printf("\n");
}
class Solution {
	vector<vector<int>> path;
	vector<int> nodes;
public:
	void pathSumHelper(TreeNode* root, int sum) {
		if (root == NULL)return;
		//cout << root->val << endl;
		//cout << sum << endl;
		//for (auto node : nodes)
		//{
		//	cout << node;
		//}
		//cout << endl;
		sum -= root->val;
		nodes.emplace_back(root->val);
		if (root->left == NULL&&root->right == NULL && 0 == sum)
		{
			path.emplace_back(nodes);
			nodes.pop_back();
			return;
		}
		// 剪枝  不行，有负数
		//if (sum<0)return;
		pathSumHelper(root->left,sum);
		pathSumHelper(root->right, sum);
		// 回溯
		nodes.pop_back();
	}
	vector<vector<int>> pathSum(TreeNode* root, int sum) {
		pathSumHelper(root, sum);
		return path;
	}
};

int main()
{
	TreeNode * root = NULL;
	CreateBiTree(root);
	Solution s;
	auto paths = s.pathSum(root,22);
	for (auto v : paths)
	{
		for (auto node:v)
		{
			cout << node;
		}
		cout << endl;
	}
	return 0;
}