码迷,mamicode.com
首页 > 其他好文 > 详细

Path Sum II

时间:2015-02-28 16:14:26      阅读:155      评论:0      收藏:0      [点我收藏+]

标签:

https://oj.leetcode.com/problems/path-sum-ii/

Given a binary tree and a sum, find all root-to-leaf paths where each path‘s sum equals the given sum.

For example:
Given the below binary tree and sum = 22,

              5
             /             4   8
           /   /           11  13  4
         /  \    /         7    2  5   1

return

[
   [5,4,11,2],
   [5,8,4,5]
]

解题思路:

这题和上题 Path Sum 很类似,在他的基础上,要求输出所有和为sum的path。上题讲过,这种题目一般用DFS来做,那么在递归DFS的时候,同时维护一个总的结果的List<List<Integer>> resultList,和一个当前可能的子结果List<Integer> currentList。

这道题本该一次性就写出来,无奈犯了一个非常低级的语言层面的错误。符合条件的时候,直接将currentList加入到resultList中了。这样resultList中等于是存了N个一模一样的list对象了。应该每次拷贝currenList生成一个新的对象放入resultList,就不会有问题了。代码如下:

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        List<List<Integer>> resultList = new ArrayList<List<Integer>>();
        List<Integer> currentList = new ArrayList<Integer>();
        DFS(root, sum, 0, resultList, currentList);
        return resultList;
    }
    
    public void DFS(TreeNode root, int sum, int sumCurrent, List<List<Integer>> resultList, List<Integer> currentList){
        if(root == null){
            return;
        }
        
        currentList.add(root.val);
        sumCurrent = sumCurrent + root.val;
        if(root.left == null && root.right == null){
            if(sumCurrent == sum){
                //之前一直错,犯了低级错误!!!resultList.add(currentList);
                resultList.add(new LinkedList(currentList));
            }
        }
        DFS(root.left, sum, sumCurrent, resultList, currentList);
        DFS(root.right, sum, sumCurrent, resultList, currentList);
        currentList.remove(currentList.size() - 1);
        //下面一行可要可不要
        // sumCurrent = sumCurrent - root.val;
    }
}

可是至少有一个问题,为什么同样是子状态,前面currentList就一定要回溯,currentSum就不要回溯?

上面的算法中,我们看到一个DFS+回溯算法的基本模样,前面提到的另外两道题DFS题目,Letter Combinations of a Phone Number 和 Word Break II 也都是这个模型。可以说,这种算法无论是在面试中还是实际编程中都是相当重要的,理解了它会有种豁然开朗的感觉。我们可以稍微总结一下。

首先,判断递归返回的条件,写在最开始。然后,处理当前节点。处理完当前节点后,DFS下一节点,最后回溯。或者循环处理当前节点内的所有可能的状态(比如上面电话号码簿里一个数字所有代表的字母,这时一般用for循环)。

回溯是这个算法的关键,下面我借用一个网友的伪代码,讲讲自己对上面的问题,也就是回溯,为什么要回溯的理解。引用的网址在下面有。

void dfs(int 当前状态)
    {
          if(当前状态为边界状态)
          {
            记录或输出
            return;
          }
          for(i=0;i<n;i++)        //横向遍历解答树所有子节点
         {
               //扩展出一个子状态。
               修改了全局变量
               if(子状态满足约束条件)
                {
                  dfs(子状态)
               }
                恢复全局变量//回溯部分
            }
    }

从上面的伪代码里我们可以看到,对当前节点的一个状态的操作,往往会包含对全局变量的修改,类似于本题中的currentList。想想递归的过程,实际上是一个树形的占空间,每个子状态都有自己的栈,存放当前的临时变量,这也就是为什么有时候递归非常耗费空间的原因。

这时我们再来看上面提出的问题,为什么同样是子状态,前面currentList就一定要回溯,currentSum就不要回溯?这里我只说Java,因为自己对C或者C++不了解。

在Java里,所有方法(或函数)的参数,只有pass by value,这和C++可以传引用是非常不同的。所以所有的primitive type,比如int, char等,传的都是值。在本题中,DFS方法的参数,sum和sumCurrent传的都是他们的数值,而不是本身,他们都是每个递归状态都会维护在自己占空间内的,所以后面的修改不会影响前面栈内的值。这就是currentSum不需要回溯的原因。用上面的伪代码来说,它不是全局变量。最后一行sumCurrent = sumCurrent - root.val;是多余的,因为它减也是减去当前栈内的值,和其他栈并不相关。

反过来,currenList是一个对象。严格来说,Java中,currentList中存的是一个内存地址,这片内存内才住着真正的对象。所以DFS方法传参的时候,pass by value,就将这个地址传了进去。在整个递归的过程中,所有对currenList的修改,都是修改的这一个内存地址,也就是这一个对象。各自自状态的栈空间里,保存的副本也仅仅是这个内存地址的值。各个递归状态对currenList的修改,都会影响其他状态。这也就是为什么currentList需要回溯的原因。dfs修改后,需要立刻恢复它到修改前的状态。

有其他网友总结的更好,我来转载下。

http://blog.csdn.net/fightforyourdream/article/details/12866861

http://blog.chinaunix.net/uid-26602509-id-3178938.html

http://blog.sina.com.cn/s/blog_779fba530100wqz9.html

http://blog.csdn.net/cdfr2321388/article/details/5725863

从这个问题上,我们还能比较好的理解Java里pass by value的实质,为什么primitive就是pass by value,而object type却感觉是pass by reference,实际上也是pass by value。

那么思路来了,我们能不能将这个currenList在每次递归的栈空间里都维护一个副本,这样就不需要回溯了嘛。下面的代码便是这样的一个实现。

/**
 * Definition for binary tree
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution {
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        List<List<Integer>> resultList = new ArrayList<List<Integer>>();
        List<Integer> currentList = new ArrayList<Integer>();
        DFS(root, sum, 0, resultList, currentList);
        return resultList;
    }
    
    public void DFS(TreeNode root, int sum, int sumCurrent, List<List<Integer>> resultList, List<Integer> currentList){
        if(root == null){
            return;
        }
        
        //回溯的关键,每次申明一个新的内存放当前状态的结果currentList,后面就不要回溯了
        List<Integer> soFarList = new LinkedList(currentList);
        soFarList.add(root.val);
        sumCurrent = sumCurrent + root.val;
        if(root.left == null && root.right == null){
            if(sumCurrent == sum){
                //之前一直错,犯了低级错误!!!resultList.add(currentList);
                resultList.add(new LinkedList(soFarList));
            }
        }
        DFS(root.left, sum, sumCurrent, resultList, soFarList);
        DFS(root.right, sum, sumCurrent, resultList, soFarList);
        //下面一步就不必要了,这个对理解回溯很重要
        // currentList.remove(currentList.size() - 1);
        //下面一行可要可不要
        // sumCurrent = sumCurrent - root.val;
    }
}

 应该说,回溯放在递归方法的最后面,理解起来是有点难度的,这样一解释应该能清楚很多。

这类题目在面试中经常出现,以及DFS和BFS的区别,啥时候用啥,各自优缺点之类,需要好好理解。

Path Sum II

标签:

原文地址:http://www.cnblogs.com/NickyYe/p/4305425.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!