【劍指offer】 34 二叉樹中和爲某一值的路徑

面試題34. 二叉樹中和爲某一值的路徑

難度中等27

輸入一棵二叉樹和一個整數,打印出二叉樹中節點值的和爲輸入整數的所有路徑。從樹的根節點開始往下一直到葉節點所經過的節點形成一條路徑。

 

示例:
給定如下二叉樹,以及目標和 sum = 22

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

返回:

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

 

提示:

  1. 節點總數 <= 10000

進行先序遍歷,再進行判斷。當前結點爲空返回,當前結點的左右子樹爲空且值爲路徑值,將路徑加入鏈表中,然後當他的左右子樹遍歷完成之後,把他從路徑鏈表中刪除。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    LinkedList<List<Integer>> list = new LinkedList<>();
    LinkedList<Integer> list1 = new LinkedList<>();
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        help(root,sum);
        return list;
    }
    public void help(TreeNode root,int sum){
        if(root == null)
            return;
        list1.add(root.val);
        sum -= root.val;
        if(sum == 0 && root.left == null && root.right == null){
            list.add(new LinkedList(list1));
        }
        
        help(root.left,sum);//左子樹遍歷完成
        help(root.right,sum);//右子樹遍歷完成
        list1.removeLast();    
        
    }
}

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章