本系列導(dǎo)航：劍指offer（第二版）java實(shí)現(xiàn)導(dǎo)航帖

面試題34：二叉樹中和為某一值的路徑

題目要求：
輸入一棵二叉樹和一個(gè)整數(shù)，打印出二叉樹中節(jié)點(diǎn)值的和為輸入整數(shù)的所有路徑。從樹的根節(jié)點(diǎn)開始往下一直到葉節(jié)點(diǎn)所經(jīng)過的節(jié)點(diǎn)形成一條路徑。

解題思路：
需要得到所有從根節(jié)點(diǎn)到葉節(jié)點(diǎn)的路徑，判斷和是否為給定值。自己寫一個(gè)小的二叉樹，通過模擬上述過程，發(fā)現(xiàn)獲取所有路徑的過程與前序遍歷類似。
因此，可以對給定二叉樹進(jìn)行前序遍歷。當(dāng)被訪問節(jié)點(diǎn)時(shí)葉節(jié)點(diǎn)時(shí)，判斷路徑和是否為給定值。此外，為了記錄路徑上的節(jié)點(diǎn)，需要一個(gè)數(shù)組。

package chapter4;
import structure.TreeNode;
import java.util.ArrayList;
import java.util.List;

/**
 * Created by ryder on 2017/7/18.
 * 二叉樹中和為某一值的路徑
 */
public class P182_FindPath {
    //用類似于前序遍歷的思路解決
    public static void findPath(TreeNode<Integer> root,int exceptedSum){
        if(root==null)
            return;
        List<Integer> path = new ArrayList<>();
        findPath(root,path,exceptedSum,0);
    }
    //curNode為將要被訪問的節(jié)點(diǎn)，還未被加入到path中
    public static void findPath(TreeNode<Integer> curNode,List<Integer> path,int exceptedSum,int currentSum){
        path.add(curNode.val);
        currentSum+=curNode.val;
        if(curNode.left!=null)
            findPath(curNode.left,path,exceptedSum,currentSum);
        if(curNode.right!=null)
            findPath(curNode.right,path,exceptedSum,currentSum);
        if(curNode.left==null && curNode.right==null && currentSum==exceptedSum)
            System.out.println(path);
        path.remove(path.size()-1) ;
    }
   
    public static void main(String[] args) {
        //            10
        //          /   \
        //         5     12
        //       /  \
        //      4    7
        TreeNode<Integer> root = new TreeNode<Integer>(10);
        root.left = new TreeNode<Integer>(5);
        root.right = new TreeNode<Integer>(12);
        root.left.left = new TreeNode<Integer>(4);
        root.left.right = new TreeNode<Integer>(7);
        findPath(root,22);
        findPath2(root,22);
    }
}

如果二叉樹的所有節(jié)點(diǎn)值都是大于0的（原題中并沒有這個(gè)條件），可以進(jìn)行剪枝。

//如果所有節(jié)點(diǎn)值均大于0，可進(jìn)行剪枝
    public static void findPath2(TreeNode<Integer> root,int exceptedSum){
        if(root==null)
            return;
        List<Integer> path = new ArrayList<>();
        findPath2(root,path,exceptedSum,0);
    }
     //curNode為將要被訪問的節(jié)點(diǎn)，還未被加入到path中
    public static void findPath2(TreeNode<Integer> curNode,List<Integer> path,int exceptedSum,int currentSum){
        path.add(curNode.val);
        currentSum+=curNode.val;
        //只有當(dāng)currentSum小于exceptedSum時(shí)需要繼續(xù)當(dāng)前節(jié)點(diǎn)的子節(jié)點(diǎn)的遍歷
        if(currentSum<exceptedSum){
            if(curNode.left!=null)
                findPath2(curNode.left,path,exceptedSum,currentSum);
            if(curNode.right!=null)
                findPath2(curNode.right,path,exceptedSum,currentSum);
        }
        //currentSum大于等于exceptedSum時(shí)可以直接停止當(dāng)前分支的遍歷，因?yàn)楫?dāng)前分支下currentSum只會(huì)越來越大，不會(huì)再有符合要求的解
        else if(currentSum==exceptedSum && curNode.left==null && curNode.right==null)
            System.out.println(path);
        path.remove(path.size()-1) ;
    }

運(yùn)行結(jié)果

[10, 5, 7]
[10, 12]
[10, 5, 7]
[10, 12]