LeetCode Algorithms 113. Path Sum II

题目难度： Medium

原题描述：

Given a binary tree and a sum, find all root-to-leaf paths where each path's sum equals the given sum.

For example:
Given the below binary tree and sum = 22,

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

return

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

题目大意：

        给你一棵二叉树，让你返回所有从根结点到叶子结点的所有数的和等于sum的路径。

解题思路：

        直接dfs整棵树，用一个vector存储走过的结点的值，注意回退时要把上一次加入的数删除，这样当走到叶子结点时，判断当前从根结点到叶子结点的和是否等于sum，如果相等，则将vector加入最终的结果中，这样一直到dfs结束为止。

时间复杂度分析：

        dfs的时间复杂度为O(n+E)，这里E = n-1，因此时间复杂度为O(n)，n为树的结点个数。

以下是代码：

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

class Solution {
public:
	void dfs(TreeNode* root , int sum, int addSum){
		v.push_back(root->val);
		addSum += root->val;
		if(addSum==sum && v.size()>0 && (!(root->left)&&!(root->right)) ){
			ans.push_back(v);
		}

		if(root->left){
			dfs(root->left, sum, addSum);
			v.pop_back();
		}
		if(root->right){
			dfs(root->right, sum, addSum);
			v.pop_back();
		}
	}

	vector< vector<int> > pathSum(TreeNode* root, int sum) {
        ans.clear();
        v.clear();
        if(root){
        	dfs(root, sum, 0);
		}
        return ans;
    }
};