为了加深对二叉树以及DFS以及回溯的熟练使用,也为了应对秋招面试的手撕代码,目前正在刷牛客的 算法面试题。加油!!!!
刷过回溯,经常写迷宫,发现二叉树的回溯有点奇怪,不需要将走过的路置1 以及 置0 的操作。
可能是刷大题刷多了,遇到这种简单的题,反而有点不适应,一开始的代码Bug挺多的,不过后来debug了几次,找到了一些错误,并改了回来。
代码:核心代码
vector<vector<int>>result;
vector<int>path;
void researchPath(treeNode* node,int sum)
{
//必须将 插入操作放在这,不然最后叶子结点与sum值不好判断
path.push_back(node->val);
sum -= node->val;
//终止条件,只要到了叶子结点就返回
if (!node->left && !node->right)
{
if (sum == 0)
{
result.push_back(path);
return;
}
return;
}
//单层递归逻辑
if (node->left)
{
researchPath(node->left, sum);
path.pop_back();//回溯
}
if (node->right)
{
researchPath(node->right, sum);
path.pop_back();//回溯
}
return;
}