刷剑指offer遇到两道题:28和64。64考察逻辑短路效应,而28是逻辑短路的一个坑。
先看简单题64:
求 1+2+...+n
,要求不能使用乘除法、for、while、if、else、switch、case等关键字及条件判断语句(A?B:C)。
本题在简单问题上做了许多限制,需要使用排除法一步步导向答案。
1+2+...+(n-1)+n 的计算方法主要有三种:平均计算、迭代、递归。
方法1:平均计算 (包含乘除,不能用)
int sumNums(int n) {
return (1 + n) * n / 2;
}
方法2:迭代 (循环必须使用 while 或 for,因此本方法不可取,直接排除)
int sumNums(int n) {
int res = 0;
for(int i = 1; i <= n; i++)
res += i;
return res;
}
方法3:递归 (终止条件需要使用 ifif ,因此本方法不可取)
思考: 除了 ifi和 switch等判断语句外,是否有其他方法可用来终止递归?
int sumNums(int n) {
if(n == 1) return 1;
n += sumNums(n - 1); //注意n是局部变量,先传给下一个sumNums函数的新局部变量,后修改。没错的
return n;
}
正确解法:逻辑运算符的短路效应,代替if条件判断。
常见的逻辑运算符有三种,即 “与 & ”,“或 || ”,“非 ! ” ;而其有重要的短路效应,如下所示:
if(A && B) // 若 A 为 false ,则 B 的判断不会执行(即短路),直接判定 A && B 为 false
if(A || B) // 若 A 为 true ,则 B 的判断不会执行(即短路),直接判定 A || B 为 true
本题需要实现 “当 n = 1时终止递归” 的需求,可通过短路效应实现。
n > 1 && sumNums(n - 1) // 当 n = 1 时 n > 1 不成立 ,此时 “短路” ,终止后续递归。
class Solution {
public:
int sumNums(int n) {
n > 1 && (n += sumNums(n - 1));
return n;
}
};
复杂度分析:
时间复杂度 O(n): 计算 n + (n-1) + ... + 2 + 1需要开启 n个递归函数。
空间复杂度 O(n): 递归深度达到 nn ,系统使用 O(n)大小的额外空间。
当然,也可以利用逻辑或||的短路效应。
class Solution {
public:
int sumNums(int n) { //注意这里的n是局部变量,每次调用子函数都有一个新的n
n==1 || (n+=sumNums(n-1));
return n;
}
};
看28题:
请实现一个函数,用来判断一棵二叉树是不是对称的。如果一棵二叉树和它的镜像一样,那么它是对称的。例如,二叉树 [1,2,2,3,4,4,3] 是对称的。
class Solution {
public:
//判断二叉树root是否对称
bool isSymmetric(TreeNode* root) {
return (root==nullptr)||isMirror(root->left,root->right);
//树对称:要么空树,要么左右子树互为镜像
}
//判断两棵二叉树是否镜像,这里是L和R即左右子树
bool isMirror(TreeNode* L,TreeNode* R){
if(L==nullptr && R==nullptr) return true;
//如果不是左右子树都为空,那么必然是3种情况:1.左不空,右空;2.左空,右不空;3.左右都不空。情况1、2肯定不是镜像!其中,情况3如果左右节点值相同,则继续判断其子树是否镜像;若左右节点值不同,则必定不是镜像!所以得到下列if条件:
if( L==nullptr || R==nullptr || L->val != R->val) return false; //条件的顺序不能变
return isMirror(L->left,R->right)&&isMirror(L->right,R->left);
}
};
值得注意的是,交换条件顺序,报错了。
if(L->val != R->val|| L==nullptr || R==nullptr) return false; //报错
为什么报错呢?如果L为nullptr,它就不存在左右孩子,这时候用L->val != R->val判断会出错!所以空子树(叶子)需要单独一个条件判断false,而且要在L->val之前判断,这样nullptr->val不会被调用,利用了||的短路!
也就是说,要保证L不为nullptr,才能调用L->val。否则报错。这提醒我们,平时写代码要注意判断操作是否合法,尤其是指针容易犯这样的错误。
需要区分的还有:二叉树的子树(必须子树完全相同)vs. 二叉树的子结构(可以截取子树一部分相同)。
这两道题其实也涉及到逻辑短路,不小心就会出错!
树的子结构可以是下面这样
参考这篇博客 https://www.cnblogs.com/chengdabelief/p/7629438.html
是否为二叉树的子结构:剑指offer 26题
class Solution {
public:
//先序遍历A的每个节点,并检查以该节点为根的子树是否包含B
bool isSubStructure(TreeNode* A, TreeNode* B) {
return (A!=nullptr && B!=nullptr) && (recur(A,B)||isSubStructure(A->left,B)||isSubStructure(A->right,B));
}
//递归函数,检查节点值是否相同
bool recur(TreeNode* A,TreeNode* B){
if(B==nullptr) return true;
if(A==nullptr || A->val != B->val) return false;
return recur(A->left,B->left) && recur(A->right,B->right);
}
};
是否为二叉树的子树:https://leetcode-cn.com/problems/check-subtree-lcci/
//递归。区分是否为子树和是否为树的子结构!!!
class Solution {
public:
bool checkSubTree(TreeNode* t1, TreeNode* t2) {
if(t2==nullptr) return true; //空树是任何树的子树,不能和下面的if换位置
if(t1==nullptr) return false; //空树没有子树
if(t1->val == t2->val){
//t1左子树是否和t2左子树完全相同&&t1右子树是否和t2右子树完全相同,也就是说t1和t2完全一样
return checkSubTree(t1->left,t2->left)&&checkSubTree(t1->right,t2->right);
}
else{
//t2是否和t1左子树的某子树完全相同||t2是否和t1右子树的某子树完全相同
return checkSubTree(t1->left,t2)||checkSubTree(t1->right,t2);
}
}
};
参考:
作者:Krahets
链接:https://leetcode-cn.com/leetbook/read/illustration-of-algorithm/9h1gyt/
来源:力扣(LeetCode)