【Java】 剑指offer(34) 二叉树中和为某一值的路径

本文参考自《剑指offer》一书,代码采用Java语言。

更多:《剑指Offer》Java实现合集  

题目 

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

思路

  1.假设找到了其中一条路径,达到叶结点后,由于没有指向父结点的指针,所以必须提前创建一个链表存储前面经过的结点。

  2.由于是从根结点出发,所以要想到使用前序遍历

  3.利用链表存储结点,在该结点完成左右子树的路径搜索后(即递归函数结束,返回到其父结点之前),要删除该结点,从而记录别的路径。

  具体实现:通过前序遍历,从根结点出发,每次在链表中存储遍历到的结点,若到达叶子结点,则根据所有结点的和是否等于输入的整数,判断是否打印输出。在当前结点访问结束后,递归函数将会返回到它的父结点,所以在函数退出之前,要删除链表中的当前结点,以确保返回父结点时,储存的路径刚好是从根结点到父结点。

  改进:书中的代码是根据所有结点的和是否等于输入的整数,判断是否打印输出。其实没有这个必要,只需要在每次遍历到一个结点时,令目标整数等于自己减去当前结点的值,若到达根结点时,最终的目标整数等于0就可以打印输出。(描述得不是很清楚,就是相当于每个结点的目标整数不同,详见代码)

测试算例 

  1.功能测试(一条或者多条对应路径;无对应路径;结点值为正负零;)

  2.特殊测试(根结点为null)

Java代码

//题目:输入一棵二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所
//有路径。从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

public class PathInTree {
	public class TreeNode {
	    int val = 0;
	    TreeNode left = null;
	    TreeNode right = null;
	    public TreeNode(int val) {
	        this.val = val;
	    }
	}

	public void findPath(TreeNode root,int target) {
		if(root==null)
			return;
		ArrayList<Integer> list= new ArrayList<>();
		printPath(root, target,list);
	}

	private void printPath(TreeNode node,int target,ArrayList<Integer> list) {
		if(node==null)
			return;
		list.add(node.val);
		target-=node.val;  //每个结点的target不会受到方法的影响而改变
		if(target==0 && node.left==null && node.right==null) {  //叶子结点
				for (Integer integer : list)
					System.out.print(integer+" ");
				System.out.println();
		}else {		//中间结点
			printPath(node.left, target, list);
			printPath(node.right, target, list);
		}
		list.remove(list.size()-1);
	}

  

  牛客网中的题目有多两点要求:1.返回类型为ArrayList<ArrayList<Integer>>,即返回所有路径的集合;2.要求返回的集合中,长度大的靠前。下面是实现的代码:

/*
 * 几个要点:
 * 1. 将nodeList和pathList定义成全局变量,避免在方法中的多次传递
 * 2. 在pathList中添加nodeList时,因为nodeList会不断变化,所以必须新建一个list存入
 *    复制ArrayList的方法:newList=new ArrayList<Integer>(oldList)(复制内容,而不是复制地址,注意与newList=oldList的区分)
 * 3. 在当前结点完成左右子树的路径搜索后,记得删除nodeList中的当前结点
 * 4. target是基本数据类型int,不会受到方法的影响而改变
 */
private ArrayList<Integer> nodeList= new ArrayList<>();
private ArrayList<ArrayList<Integer>> pathList = new ArrayList<>();

public ArrayList<ArrayList<Integer>> FindPath(TreeNode node,int target) {
    if(node==null)
        return pathList;
    nodeList.add(node.val);
    target-=node.val;
    if(target==0 && node.left==null && node.right==null) {  //叶子结点
        //长度大的nodeList插入到pathList的前面
        int i=0;
        while(i<pathList.size() && nodeList.size()<pathList.get(i).size() ) //记得防止越界
                i++;
        pathList.add(i, new ArrayList<Integer>(nodeList));  //nodeList会随方法变化,必须新建一个list存入pathList中!
    }else {  //不是叶子结点
        pathList=FindPath(node.left, target);
        pathList=FindPath(node.right, target);
    }
    nodeList.remove(nodeList.size()-1);  //记得删除当前结点
    return pathList;
}

  

收获

  1.二叉树的许多题目与遍历(包括层序遍历)有关,要深刻理解;根据根结点的位置判断使用哪一种遍历。

  2.二叉树遍历过程没有父结点指针,要保存路径的话,必须要创建容器存储之前的结点。

  3.熟悉这道题中在每次递归函数结束前删除当前结点的操作,这可以确保返回到父结点时路径刚好是从根结点到父结点。

  4.target-=node.val这句代码非常好,多多体会。

  5.牛客网的那部分代码:在链表中存储一个对象时,如果该对象是不断变化的,应该创建一个新的对象复制该对象的内容(而不是指向同一个对象),将这个新的对象存储到链表中。基本数据类型和String则没有这种问题。说到底其实是存储的是地址还是值的问题。这篇文章讨论了一下。

更多:《剑指Offer》Java实现合集  

  

原文地址:https://www.cnblogs.com/yongh/p/9821754.html

时间: 2024-08-28 01:04:28

【Java】 剑指offer(34) 二叉树中和为某一值的路径的相关文章

【剑指Offer】二叉树中和为某一值的路径

问题描述 输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径.路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径. 算法分析 如果 root 等于 输入数,将root放在返回数组里返回:如果root大于输入数,返回空值: 如果root小于输入数,将root放在数组里,输入数自减root,一同随root的子树递归: 如果输入数为0了且左右子树都为空,即为叶子节点,则这条路径可行,返回数组: 如果 root 没有子树了,同时输入数还没为0,说明此路不通,返回NUL

剑指OFFER之二叉树中和为某一值的路径

题目描述: 输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径.路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径. 输入: 每个测试案例包括n+1行: 第一行为2个整数n,k(1<=n<=10000),n表示结点的个数,k表示要求的路径和,结点编号从1到n. 接下来有n行.这n行中每行为3个整数vi,leftnode,rightnode,vi表示第i个结点的值,leftnode表示第i个结点的左孩子结点编号,rightnode表示第i个结点的右孩子结点编号

python剑指offer系列二叉树中和为某一值的路径

题目描述 输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径.路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径.(注意: 在返回值的list中,数组长度大的数组靠前) # -*- coding:utf-8 -*- # class TreeNode: # def __init__(self, x): # self.val = x # self.left = None # self.right = None class Solution: # 返回二维列

剑指offer:二叉树中和为某一值的路径

题目描述输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径.路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径.(注意: 在返回值的list中,数组长度大的数组靠前) class TreeNode: def __init__(self, x): self.val = x self.left = None self.right = None class Solution: # 返回二维列表,内部每个列表表示找到的路径 """ 根据题目

LeetCode | 面试题34. 二叉树中和为某一值的路径【剑指Offer】【Python】

LeetCode 面试题34. 二叉树中和为某一值的路径[剑指Offer][Medium][Python][回溯] 问题 力扣 输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径.从树的根节点开始往下一直到叶节点所经过的节点形成一条路径. 示例: 给定如下二叉树,以及目标和 sum = 22, 5 / 4 8 / / 11 13 4 / \ / 7 2 5 1 返回: [ [5,4,11,2], [5,8,4,5] ] 提示: 节点总数 <= 10000 注意:本题与主站 1

剑指Offer-- 二叉搜索树中和为某一值的路径

输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径.路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径. 本身题目不是很难,但是因为刚接触pyhon,对一些对象的传递不太了解,所以跳了很久也没有通过,后来看到这篇文章  后才明白,犯了一样的错误 # -*- coding:utf-8 -*- # class TreeNode: # def __init__(self, x): # self.val = x # self.left = None # self.ri

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

class Solution: def pathSum(self, root: TreeNode, sum: int) -> List[List[int]]: res=[] def traceback(node,trace,sum): if not node: return if node.val==sum and (not node.left) and (not node.right): res.append(trace+[node.val]) return if node.left: tra

二叉树中和为某一值的路径-剑指Offer

二叉树中和为某一值的路径 题目描述 输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径.路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径. 思路 用栈来存储二叉树的路径,遍历到叶子结点就判断是否总和为target,是的话打印,若不是叶子结点,那继续遍历左右子树 注意:打印完后返回父结点时要弹出栈顶的叶子结点,并且总和减去该值 注意:打印路径时要遍历栈 代码 import java.util.ArrayList; import java.util.Stack;

剑指OFFER之二叉树的镜像(九度OJ1521)

题目描述: 输入一个二叉树,输出其镜像. 输入: 输入可能包含多个测试样例,输入以EOF结束.对于每个测试案例,输入的第一行为一个整数n(0<=n<=1000,n代表将要输入的二叉树节点的个数(节点从1开始编号).接下来一行有n个数字,代表第i个二叉树节点的元素的值.接下来有n行,每行有一个字母Ci.Ci='d'表示第i个节点有两子孩子,紧接着是左孩子编号和右孩子编号.Ci='l'表示第i个节点有一个左孩子,紧接着是左孩子的编号.Ci='r'表示第i个节点有一个右孩子,紧接着是右孩子的编号.C