题目:二叉树中和为某一值的路径
题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
解题思路
当用前序遍历的方式访问到某一节点时,我们把该节点添加到路径上,并累加该节点的值。如果该节点为叶节点,并且路径中节点值的和刚刚好等于输入的整数,则当前路径符合要求,然后打印该路径。如果当前节点不是叶节点,那么继续访问它的子节点。当前节点访问结束之后,递归函数将自动回到它的父节点。因此,我们在函数退出之前需要在路径上删除当前节点并减去当前节点的值,以确保返回父节点时路径刚好是从根节点到父节点。
参考代码
using System.Collections.Generic;
/*
public class TreeNode
{
public int val;
public TreeNode left;
public TreeNode right;
public TreeNode (int x)
{
val = x;
}
}*/
class Solution
{
public List<List<int>> FindPath(TreeNode root, int expectNumber)
{
// write code here
List<List<int>> list = new List<List<int>>();
List<int> path = new List<int>();
//如果根节点不为空,那么就遍历树的所有节点查找是否有符合要求的路径
if (root != null)
DFS(root,expectNumber,ref list,ref path);
return list;
}
private void DFS(TreeNode root,int sum,ref List<List<int>> list,ref List<int> path)
{
//把当前节点添加进集合
path.Add(root.val);
//如果当前节点为叶节点,且和与期待值相等,则把路径添加到路径集合中
if (root.left == null && root.right == null)
{
//这里因为path的元素会随着递归的改变而发生变化,所以在储存到路径集合list里时需要开辟新的空间储存路径值再保存
if (sum - root.val == 0)
list.Add(new List<int>(path));
path.Remove(root.val);
return;
}
//如果当前节点有子节点,那么递归进入子节点,传入的期待值必须减去当前节点值
if (root.left != null)
DFS(root.left, sum - root.val, ref list, ref path);
if (root.right != null)
DFS(root.right, sum - root.val, ref list, ref path);
//当遍历完所有子节点回到上一级函数的时候,需要把当前的节点移出集合
path.Remove(root.val);
}
}