题意需要记录所有可能的路径,可以设置一个变量path用于存储每次产生的路径,设置一个变量result存储所有符合条件的路劲。
用递归可以实现,主要注意的问题是:对path路径的pop操作。
方式一:每一次深度遍历之后,得到一条路径,符合条件就压入result中,然后要判断下一天路径。下一条路径是将前一条路径中的最后一个左节点弹出,压入右节点。下面这种方式,就是在寻得叶子节点之后,继续执行叶子节点的左节点和右节点的遍历,两种都是return,之后执行的pop操作,就可以弹出最后压入的叶子节点。
class Solution { public: vector<vector<int> > pathSum(TreeNode *root, int sum) { vector<vector<int> > result; vector<int> path; if(!root) return result; getPath(result,path,root,sum); return result; } private: void getPath(vector<vector<int> > &result,vector<int> &path,TreeNode *root,int sum) { if(!root) return ; path.push_back(root->val); if(root->left ==nullptr&&root->right==nullptr&&root->val==sum) { result.push_back(path); //return; } getPath(result,path,root->left,sum-root->val); //path.pop_back(); getPath(result,path,root->right,sum-root->val); path.pop_back(); } };
方式二:
单独处理叶子节点,遇到叶子节点则不再往下遍历,将其pop出,继续下一次右节点的遍历
class Solution { public: vector<vector<int> > pathSum(TreeNode *root, int sum) { vector<vector<int> > result; vector<int> path; if(!root) return result; getPath(result,path,root,sum); return result; } private: void getPath(vector<vector<int> > &result,vector<int> &path,TreeNode *root,int sum) { if(!root) return ; path.push_back(root->val); if(root->left ==nullptr&&root->right==nullptr&&root->val==sum) { result.push_back(path); path.pop_back(); return; } if(root->left ==nullptr&&root->right==nullptr&&root->val!=sum) { path.pop_back(); return; } getPath(result,path,root->left,sum-root->val); //path.pop_back(); getPath(result,path,root->right,sum-root->val); path.pop_back(); } };
时间: 2024-10-10 11:38:51