剑指offer 34 二叉树中找路径

输入一棵二叉树和一个整数，打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。

这道题用C做实在是太麻烦了, 换了C++好清爽!

很容易就能想到递归, 思路难度也不大

几个坑: 必须是从root到leaf的完整路径, 在非叶结点的时候满足要求是不可以的!!
val可能为负数, 因此也可能导致在中间就出现了满足sum

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> result;

    vector<vector<int>> pathSum(TreeNode* root, int sum) {
        vector<int> arr;
        if(root) check(root, sum, arr);
        return result;
    }

    void check(TreeNode *root, int sum, vector<int> arr)
    {        
        sum -= root->val;
        //叶结点, 且符合结果
        if(!root->left && !root->right && !sum)
        {
            arr.push_back(root->val);
            result.push_back(arr);
            return;
        }
        //非叶节点, 且结果不为0
        else if((root->left || root->right))
        {
            arr.push_back(root->val);
            if(root->left) check(root->left, sum, arr);
            if(root->right) check(root->right, sum, arr);
        }

        return;

    }
};

时间复杂度: logN
空间复杂度 logN