Leetcode 剑指系列 Day 19 搜索与回溯
1.剑指 Offer 64. 求1+2+…+n
解题思路:利用逻辑运算符的短路效应进行计算
class Solution {
public:
int add(int n){
int temp = 0;
n && (temp = add(n - 1));//若n = 0,那么temp = add(n - 1)不会执行
return n + temp;
}
int sumNums(int n) {
return add(n);
}
};
2.剑指 Offer 68 - I. 二叉搜索树的最近公共祖先
解题思路:根据二叉搜索树的性质,根据节点值逐个向下遍历即可,设valb为p,q值的较大值,vals为p,q值的较小值,若当前节点值小于vals则向右下子树遍历,若大于valb则向左下子树遍历,否则退出循环并返回
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
int valb = p->val > q->val ? p->val : q->val;
int vals = p->val > q->val ? q->val : p->val;
TreeNode* node = root;
while(1){
if(node->val < vals) node = node->right;
else if(node->val > valb) node = node->left;
else break;
}
return node;
}
};
3.剑指 Offer 68 - II. 二叉树的最近公共祖先
解题思路:采用后序遍历,并设置两个标记位,遍历至p把标记位1至为1,遍历至q把标记位2置为1,当两个标记位均为1的时候该节点即为两节点的公共节点。
算法过程:
终止条件:当前节点为空或者与p、q相等,返回当前节点,即root
后序遍历:
遍历左孩子、右孩子并拿两个临时节点l、r保存
若l、r均不为空,即左右孩子都检测到了p、q,那么当前节点即为最近祖先
若l不为空,r为空,应当返回l
若r不为空,l为空,应当返回r
否则,返回NULL
class Solution {
public:
TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
if(root == NULL || root == p || root == q) return root;
TreeNode* l = lowestCommonAncestor(root->left, p, q);
TreeNode* r = lowestCommonAncestor(root->right, p, q);
if(l != NULL && r != NULL) return root;
if(l == NULL && r != NULL) return r;
if(l != NULL && r == NULL) return l;
return NULL;
}
};