二叉树中和为某一值的路径

题目描述

输入一颗二叉树和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

解题思路

用前序遍历的方式访问到某一结点时，把该结点添加到路径上，并用目标值减去该节点的值。如果该结点为叶结点并且目标值减去该节点的值刚好为0，则当前的路径符合要求，我们把加入res数组中。如果当前结点不是叶结点，则继续访问它的子结点。当前结点访问结束后，递归函数将自动回到它的父结点。因此我们在函数退出之前要在路径上删除当前结点，以确保返回父结点时路径刚好是从根结点到父结点的路径。

参考代码

package 剑指offer;

import java.util.ArrayList;

import 剑指offer.重建二叉树.TreeNode;
/**
 * Definition for binary tree 
 * public class TreeNode { 
 * int val = 0; 
 * TreeNode left = null;
 * TreeNode right =null; 
 * TreeNode(int x){ val = x; } 
 *  }
 */

public class 二叉树中和为某一值的路径 {
	ArrayList<ArrayList<Integer>> res = new ArrayList<>();
	ArrayList<Integer> temp = new ArrayList<>();

	public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target){
		if(root == null)
			return res;
		target -= root.val;
		temp.add(root.val);
		if(target == 0 && root.left == null && root.right == null)
			res.add(new ArrayList<Integer>(temp));
			//res.add(temp);    引用，后面temp改变    res存储的值也会变化,所有这不可取
		else{
			FindPath(root.left, target);
			FindPath(root.right, target);
		}
		temp.remove(temp.size()-1);
		return res;
	}

	public static void main(String[] args) {
		二叉树中和为某一值的路径 a = new 二叉树中和为某一值的路径();
		TreeNode q = new TreeNode(8);
		q.left = new TreeNode(7);
		q.right = new TreeNode(9);
		ArrayList<ArrayList<Integer>> res = a.FindPath(q, 17);
		System.out.println(res.size());
		for (ArrayList<Integer> arrayList : res) {
			System.out.println(arrayList);
		}
	}

}

心得

1.res作为保存所有路径的集合，定义为全局变量
2.temp作为保存一条完整的路径，也定义为全局，为了防止前一条路径的值影响后一条，在函数的最后删除之前添加的节点
3.由于temp是作为全局变量，在函数中引用作为保存路径的集合，所有在添加到res中时需要重新new一个数组，避免引用的res在函数调用结束全部为空的情况。