二叉树解法记录:
由于二叉树的算法题内容太多了,在这记录一下:
** 递归 **
一种通用的递归方式:
例子:路经总和题目:
class Solution {
public:
void AddSum(TreeNode* node,int &sum,bool& res,int targetSum){//问题就是传入参数太多
if(!node){return; }
sum+=node->val;
if(sum==targetSum&&!node->left&&!node->right){
res = true;
}
if(node->left){
AddSum(node->left,sum,res,targetSum);
sum-=node->left->val;}
if(node->right){
AddSum(node->right,sum,res,targetSum);
sum-=node->right->val;}
}
bool hasPathSum(TreeNode* root, int targetSum) {
int sum = 0;
bool res = false;
AddSum(root,sum,res,targetSum);
return res;
}
};
作者:i3rave-montalcininka
链接:https://leetcode.cn/problems/path-sum/solution/lu-jing-zong-he-zhi-wo-zhong-yu-xue-hui-0x0db/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
卡哥的bool遍历,简洁但是比较难懂。
class Solution {
private:
bool traversal(TreeNode* cur, int count) {
if (!cur->left && !cur->right && count == 0) return true; // 遇到叶子节点,并且计数为0
if (!cur->left && !cur->right) return false; // 遇到叶子节点直接返回
if (cur->left) { // 左
count -= cur->left->val; // 递归,处理节点;
if (traversal(cur->left, count)) return true;
count += cur->left->val; // 回溯,撤销处理结果
}
if (cur->right) { // 右
count -= cur->right->val; // 递归,处理节点;
if (traversal(cur->right, count)) return true;
count += cur->right->val; // 回溯,撤销处理结果
}
return false;
}
public:
bool hasPathSum(TreeNode* root, int sum) {
if (root == NULL) return false;
return traversal(root, sum - root->val);
}
};
模板:对于要在遍历过程中改变的,比如bool,经过的路径等,采用引用传入,不变的比如比较的target,采用形参传入。
void FuncName(argv 1,argv &2){
//logic
}
二叉树的题目大多都是需要递归完成,递归类题目要记住开始,中间,结束三部分,即对于一个递归函数的传入参数和开始条件,中间的处理逻辑,终止条件。
递归的传参方式:
1.找到一个就算的:一般使用bool类型进行传递,找到了直接return。在逻辑中,如果子节点返回true证明满足条件了,再进行下一步的判断。
2.要找到全部的,一般是void类型,然后把需要修改的变量引用传入
3.递归的子节点会影响父亲节点的值的,一般用int
如果需要搜索整棵二叉树且不用处理递归返回值,递归函数就不要返回值。(这种情况就是本文下半部分介绍的113.路径总和ii)
如果需要搜索整棵二叉树且需要处理递归返回值,递归函数就需要返回值。 (这种情况我们在236. 二叉树的最近公共祖先 (opens new window)中介绍)
如果要搜索其中一条符合条件的路径,那么递归一定需要返回值,因为遇到符合条件的路径了就要及时返回。(本题的情况)
二叉树的所有题目基本都可以认为是遍历的一种特殊应用,如比较的题目:递归遍历所有的节点进行比较。反转二叉树:对每个节点进行反转。
1.遍历类:
顺序遍历和层序遍历:
前中后序遍历:
前:中左右
中:左中右
后:右中左
层序遍历: 一般不是递归
利用队列 ,一般考虑每层的处理时候使用,并且对处理逻辑进行一些调整。
如右视图树,最左下角等
2.迭代类:
全部的递归过程都可以用栈或者队列完成,即递归的实现原理也是程序栈的调用。
前中后序遍历:
前:中进,然后中出,然后右进,左进(这样符合中左右的顺序),一个节点出栈,其子节点要进栈
中:左中右
后:右中左
层序遍历:
3.比较类:
一般是比较两个树是否是一样的,比如找子树也是在这个树中找和subroot一样的一个树。
解法:一般是递归比较,递归传参为bool型,即没找到true就递归向下找。
注意对称树是比较每个节点左的左,右的右(外)和左的右,右的左(内)两个方向。
4.深度/高度类:
递归传参,return当前节点的高度/深度。然后父亲节点选择最大/最小的高度加一作为自己的高度。
5.个数类:
确定当前树有没有特点,利用特点(例如完全二叉树的个数和深度的关系)
或者递归遍历
6.平衡二叉树
其实本质也是return int的,但是其比较特殊,因为如果返回目前节点的高度,那么没法判断是否平衡并且跳出递归。如果穿目前节点的是否平衡,则没法累计高度差。
因此采用-1标记,如果已经不平衡,则返回-1.否则返回高度差,
处理逻辑:有一个子节点返回-1的直接返回-1,继续向上,否则选择返回当前节点的高度。
7.所有路径(回溯类)
也是递归的思路,但是满足递归条件时要向上一步,即返回到上一个状态,即需要一个变量来存储上一个状态。
如路径题目中:
找到最后的节点,然后此时path中为1 2 5,此时要pop_back,即表示目前的状态为2,但是2也是1的左,即对于1来说,其左不为空,因此递归在左里面找,找到了然后2也要回溯到1。
8.求和类
求和类的本质也是遍历,对于满足条件的进行求和,如左叶子,