算法训练营第二十三天 | LeetCode 530 二叉搜索树的最小绝对差、LeetCode 501 二叉搜索树中的众数、LeetCode 236 二叉树的最近公共祖先

LeetCode 530 二叉搜索树的最小绝对差

前序遍历+快排+一重遍历

class Solution {
public:
    int getMinimumDifference(TreeNode* root) {
        vector<int> nums;
        TreeNode* cur = root;
        stack <TreeNode*> mysta;
        while (cur || !mysta.empty()) {
            while (cur) {
                nums.push_back(cur->val);
                mysta.push(cur);
                cur = cur->left;
            }
            cur = mysta.top();
            mysta.pop();
            cur = cur->right;
        }
        sort(nums.begin(), nums.end());
        int num = 100000;
        for (int i = 0; i < nums.size() - 1; i++) {
            if (nums[i + 1] != nums[i] && num > nums[i + 1] - nums[i]) num = nums[i + 1] - nums[i];
        }
        return num;
    }
};

LeetCode 501 二叉搜索树中的众数

没达到进阶要求，用了栈进行了中序遍历，条件更新自己写得也比较混乱，明天再改善下。

class Solution {
public:
    vector<int> findMode(TreeNode* root) {
        vector<int> ans;
        if (!root) return ans;
        int maxCount = 0;
        int base = 100001;
        int count = 0;
        TreeNode* cur = root;
        stack<TreeNode*> mysta;
        while (cur || !mysta.empty()) {
            while (cur) {
                mysta.push(cur);
                cur = cur->left;
            }
            cur = mysta.top();
            mysta.pop();
            if (cur->val == base) {
                ++count;
            } else {
                count = 1;
                base = cur->val;
            }
            if (count == maxCount) {
                ans.push_back(base);
            }
            if (count > maxCount) {
                maxCount = count;
                ans = vector<int> {base};
            }
            cur = cur->right;
        }
        return ans;
    }
};

LeetCode 236 二叉树的最近公共祖先

这题有三指针的味道了，首先root指针指向当前节点值，由于我们要采用递归法来解决这道问题，当我们遍历到最底层的时候，有三种情况可以返回：一、是空节点；二、是p；三、是q，这样我们就得到了一个指向nullptr或者p或者q的指针；我们在函数内部还要造两个指针，一个是left指针，指向递归调用该函数传入当前root指针左节点指针作为参数的返回值，另一个是right指针，指向递归调用该函数传入当前root指针右节点指针作为参数的返回值，这样返回回来的结果，要么为空，要么就说明它指向的是p或者q了。

接下来我们再来看返回条件怎么写，当我们找到了p或者q的路径，或者就是p和q本身，前者是left指针和right指针作为返回值，后者是root作为返回值。后者已经作为递归结束条件写在开头，我们还要处理找到了p或者q的路径的情况。找到了p或者q路径的情况有这两种：一、由于我们一找到p或者q就返回，不遍历其子树，所以若它们在一条路径上，那么left和right中就只会有一个不为空；二、它们不在同一条路径上，当left指针和right指针刚好都不为空时说明找到了最近的公共节点，直接返回root指着作为上一层函数left指针或者right返回值即可，这时这个指向最近公共节点的指针也必定不为空且只在上一层节点一边。

所以返回条件可以这么写：

        if (!left && !right) return nullptr;
        if (!left) return right;
        if (!right) return left;
        return root;

而三指针同时也是满足前序遍历的一种写法，先遍历当前节点，再遍历左子树，后遍历右子树，不同的是一遇到p节点和q节点就直接返回，不对它们在同一条路径上的情况做细分处理。因为题给条件已经明确说明肯定会有符合条件的解了。

代码如下：

class Solution {
public:
    TreeNode* lowestCommonAncestor(TreeNode* root, TreeNode* p, TreeNode* q) {
        if (root == nullptr || root == p || root == q) return root;
        TreeNode* left = lowestCommonAncestor(root->left, p, q);
        TreeNode* right = lowestCommonAncestor(root->right, p, q);
        if (!left && !right) return nullptr;
        if (!left) return right;
        if (!right) return left;
        return root;
    }
};

很简单但要理解的成分比较多。不理解很难写得全面。