剣はOffer-24.二叉木の中である値の経路を指す(C+/Java)

4681 ワード

タイトル:
ツリーのヒールノードと整数を入力し、ツリーのノード値と入力整数のすべてのパスを印刷します.パスは、ツリーのルートノードからリーフノードまで続くノードがパスを形成するように定義されます.(注:戻り値のリストでは、配列長の大きい配列が先頭になります)
分析:
LeetCodeには同じタイトルがあり、以前はLeetCode 113.Path Sum IIパス総和II(C++)と記録されていたが、説明は省略された.
プログラム:
C++
class Solution {
public:
    vectorint> > FindPath(TreeNode* root,int expectNumber) {
        vector<int> v;
        dfs(root, v, expectNumber);
        return res;
    }
    void dfs(TreeNode* root, vector<int> &v, int sum){
        if(root == nullptr)
            return;
        v.push_back(root->val);
        if(root->left == nullptr && root->right == nullptr){
            if(sum == root->val)
                res.push_back(v);
        }
        else{
            dfs(root->left, v, sum-root->val);
            dfs(root->right, v, sum-root->val);
        }
        v.pop_back();
    }
private:
    vectorint>> res;
};

Java
import java.util.ArrayList;
/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

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

    }

}
*/
public class Solution {
    public ArrayList> FindPath(TreeNode root,int target) {
        res = new ArrayList<>();
        ArrayList l = new ArrayList<>();
        dfs(root, l, target);
        return res;
    }
    public void dfs(TreeNode root, ArrayList l, int sum){
        if(root == null)
            return;
        l.add(root.val);
        if(root.left == null && root.right == null){
            if(root.val == sum){
                res.add(new ArrayList(l));
            }
        }
        else{
            dfs(root.left, l, sum-root.val);
            dfs(root.right, l, sum-root.val);
        }
        l.remove(l.size()-1);
    }
    private ArrayList> res;
}