寫不出來的深度優先搜尋----leetcode113 路徑總和

一道簡單的深度優先搜尋的題目:leetcode

給定一個二叉樹和一個目標和，找到所有從根節點到葉子節點路徑總和等於給定目標和的路徑。

說明: 葉子節點是指沒有子節點的節點。

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

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:

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

直接上程式碼,用程式碼來解釋

	List<List<Integer>> res = new LinkedList<List<Integer>>(); // 全域性變數,記錄結果
    Deque<Integer> path = new LinkedList<Integer>(); // 全域性變數,記錄路徑
    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        dfs(root,sum);
        return res;
    }

    private void dfs(TreeNode node, int sum){
        if(node == null) return ; // 迴圈中止條件
        path.offerLast(node.val); // 入隊
        sum -= node.val;
        // 如果是葉子結點,且總和等於 sum,記錄結果
        if(node.left == null && node.right == null && sum == 0){
            res.add(new LinkedList(path));
        }
        dfs(node.left, sum);
        dfs(node.right, sum);
        path.pollLast(); // 出隊
    }

當然,這道題目也可以用廣度優先演算法來處理,有興趣可以試下,這裡直接給出官方題解

	List<List<Integer>> ret = new LinkedList<List<Integer>>();
    Map<TreeNode, TreeNode> map = new HashMap<TreeNode, TreeNode>();

    public List<List<Integer>> pathSum(TreeNode root, int sum) {
        if (root == null) {
            return ret;
        }

        Queue<TreeNode> queueNode = new LinkedList<TreeNode>();
        Queue<Integer> queueSum = new LinkedList<Integer>();
        queueNode.offer(root);
        queueSum.offer(0);

        while (!queueNode.isEmpty()) {
            TreeNode node = queueNode.poll();
            int rec = queueSum.poll() + node.val;

            if (node.left == null && node.right == null) {
                if (rec == sum) {
                    getPath(node);
                }
            } else {
                if (node.left != null) {
                    map.put(node.left, node);
                    queueNode.offer(node.left);
                    queueSum.offer(rec);
                }
                if (node.right != null) {
                    map.put(node.right, node);
                    queueNode.offer(node.right);
                    queueSum.offer(rec);
                }
            }
        }

        return ret;
    }

    public void getPath(TreeNode node) {
        List<Integer> temp = new LinkedList<Integer>();
        while (node != null) {
            temp.add(node.val);
            node = map.get(node);
        }
        Collections.reverse(temp);
        ret.add(new LinkedList<Integer>(temp));
    }

程式碼不難理解,但理解是一回事,能寫出來,是另外一回事