力扣112. 路径总和（递归、迭代）

力扣112. 路径总和（递归、迭代）

https://leetcode-cn.com/problems/path-sum/

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

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

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

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

 

方法1：递归

复杂度分析

时间复杂度：我们访问每个节点一次，时间复杂度为 O(N) ，其中 N 是节点个数。
空间复杂度：最坏情况下，整棵树是非平衡的，例如每个节点都只有一个孩子，递归会调用 N 次（树的高度），因此栈的空间开销是 O(N) 。但在最好情况下，树是完全平衡的，高度只有 log(N)，因此在这种情况下空间复杂度只有 O(log(N)) 。

最直接的方法就是利用递归，遍历整棵树：如果当前节点不是叶子，对它的所有孩子节点，递归调用 hasPathSum 函数，其中 sum 值减去当前节点的权值；如果当前节点是叶子，检查 sum 值是否为 0，也就是是否找到了给定的目标和。

#include "stdafx.h"
#include<queue>
#include<vector>
#include <iostream>
using namespace std;
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)
	{
		//空结点，返回false
		if (root == nullptr)return false;
		//到达叶结点，判断目标值是否相等
		//记得是要减去当前的值，再去比较
		if (root->left == nullptr && root->right == nullptr)
		{
			//等于目标值，找到了
			if (sum - root->val == 0)
			{
				return true;
			}
			else
			{
				return false;
			}
		}
		return hasPathSum(root->left, sum - root->val) || hasPathSum(root->right, sum - root->val);
	}
};
int main()
{
	TreeNode p[11] = { 2,10,6,9,11,3,7,8,4,5,12 };
	p[0].left = &p[1]; p[0].right = &p[2];

	p[1].left = &p[3]; p[1].right = &p[4];
	p[2].left = &p[5]; p[2].right = &p[6];

	p[3].left = &p[7]; p[3].right = &p[8];
	p[4].left = &p[9];
	p[5].right = &p[10];
	Solution s;
	auto result = s.hasPathSum(p, 15);
	return 0;
}

其实，做为树的递归题目是非常有套路可循的，因为树有两个分支，所以在递归里也有两个分支，一般是通过 递归 A（||，&&）递归 B 来实现分支的。只要明白了这一点，递归函数就不会很难设计。

 

方法 2：迭代

所以我们从包含根节点的栈开始模拟，剩余目标和为 sum - root.val

然后开始迭代：弹出当前元素，如果当前剩余目标和为 0 并且在叶子节点上返回 True；如果剩余和不为零并且还处在非叶子节点上，将当前节点的所有孩子以及对应的剩余和压入栈中。

 

复杂度分析

时间复杂度：和递归方法相同是 O(N)。
空间复杂度：当树不平衡的最坏情况下是 O(N) 。在最好情况（树是平衡的）下是 O(logN)。