题目:
分析:
如果做题不认真,不思考,还不如不做。
本题和主站上有一道题是一样的,测试后发现自己犯了和上次一样的一个错误。
如果到NULL才去判断,那么会导致:
1.可能有一个子树的结点会被放进去。
2.满足条件的被放两次。
特错大错的代码:
/**
* 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> > vv;
void f(TreeNode* root, int sum,vector<int> v)
{
if(root==NULL)
{
if(sum==0) vv.push_back(v);
return ;
}
v.push_back(root->val);
f(root->left,sum-root->val,v);
f(root->right,sum-root->val,v);
}
vector<vector<int>> pathSum(TreeNode* root, int sum) {
f(root,sum,vector<int> ());
return vv;
}
};
修改后的代码:
改的还行,不过改了三次才对,面试肯定又凉了。
/**
* 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> > vv;
void f(TreeNode* root, int sum,vector<int> v)
{
if(root==NULL) return ;
if(root->left==NULL && root->right==NULL)
{
if(sum==root->val)
{
v.push_back(root->val);
vv.push_back(v);
}
return ;
}
v.push_back(root->val);
f(root->left,sum-root->val,v);
f(root->right,sum-root->val,v);
}
vector<vector<int>> pathSum(TreeNode* root, int sum) {
f(root,sum,vector<int> ());
return vv;
}
};