您的位置:首页 > 职场人生

剑指Offer(第二版)面试题34:二叉树中和为某一个值的路径

2017-06-22 11:28 281 查看
(尊重劳动成果,转载请注明出处:http://blog.csdn.net/qq_25827845/article/details/73565502冷血之心的博客)

剑指Offer(第二版)面试题34:二叉树中和为某一个值的路径
题目:[b]二叉树中和为某一个值的路径[/b]
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
思路:参考剑指Offer1、当用前序遍历的方式访问到某一个节点时,我们把该节点添加到路径上,并累加该节点的值。
2、如果该节点为叶节点,并且路径中节点值的和刚好等于输入的整数,则当前路径符合要求。
3、如果当前节点不是叶节点,则继续访问它的子节点。
4、当前节点访问结束后,递归函数将自动回到它的父节点。所以,在函数退出之前要在路径上删除当前节点并减去当前节点的值,确保返回父节点时路径刚好是从根节点到父节点。

/*
* 剑指Offer面试题34:二叉树中和为某一个值的路径
* 题目:输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。
* 路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
*/
import java.util.*;

public class Solution {
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if (root == null)
return new ArrayList<ArrayList<Integer>>();
ArrayList<ArrayList<Integer>> arrayList = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> path = new ArrayList<Integer>(); // 用做存储路径
int currentSum = 0; // 当前值,用来和target比较大小
FindPath(root, target, arrayList, path, currentSum);

return arrayList;
}
public void FindPath(TreeNode root,int target, ArrayList<ArrayList<Integer>> arrayList, ArrayList<Integer> path, int currentSum) {
currentSum += root.val;
path.add(root.val);
// 如果是叶结点,并且路径和符合条件
boolean isLeaf = ((root.left==null) && (root.right==null));
if (currentSum==target && isLeaf) {
Iterator<Integer> iterator = path.iterator();
ArrayList<Integer> pathTemp = new ArrayList<>();
while (iterator.hasNext()) {
pathTemp.add(iterator.next());
}

arrayList.add(pathTemp);
}
//如果不是叶结点,遍历它的子节点
if (root.left != null)
FindPath(root.left, target, arrayList, path, currentSum);
if (root.right != null)
FindPath(root.right, target, arrayList, path, currentSum);
// 返回父节点前,在路径上删除当前节点
path.remove(path.size()-1);
}
}

class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;

public TreeNode(int val) {
this.val = val;

}

}

再来看一下大牛写的代码:
public class Solution {
private ArrayList<ArrayList<Integer>> listAll = new ArrayList<ArrayList<Integer>>();
private ArrayList<Integer> list = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root == null) return listAll;
list.add(root.val);
target -= root.val;
if(target == 0 && root.left == null && root.right == null)
listAll.add(new ArrayList<Integer>(list));
FindPath(root.left, target);
FindPath(root.right, target);
list.remove(list.size()-1);
return listAll;
}
}
完美,虽然思路一致,但是这代码咋看咋舒服


可能有人会问,按照思路中的第四条,为什么不在list.remove(list.size()-1)后边加一行target+=roo.val ?也就是使得target加上当前节点的值。
答案:递归的特性,在内层递归中对target所做的操作,并不会影响外层target的值。所以此处的target和外层递归的target互不影响哦~

牛客网AC地址:https://www.nowcoder.com/practice/b736e784e3e34731af99065031301bca?tpId=13&tqId=11177&tPage=2&rp=2&ru=/ta/coding-interviews&qru=/ta/coding-interviews/question-ranking

如果对你有帮助,记得点赞哦~欢迎大家关注我的博客,可以进群366533258一起交流学习哦~

本群给大家提供一个学习交流的平台,内设菜鸟Java管理员一枚、精通算法的金牌讲师一枚、Android管理员一枚、蓝牙BlueTooth管理员一枚、Web前端管理一枚以及C#管理一枚。欢迎大家进来交流技术。
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: