代码随想录阅读笔记-二叉树【二叉搜索树的最小绝对差】

题目

给你一棵所有节点为非负值的二叉搜索树，请你计算树中任意两节点的差的绝对值的最小值。

示例：

提示：树中至少有 2 个节点。

思路 

递归

需要用一个pre节点记录一下cur节点的前一个节点。

如图：

一些同学不知道在递归中如何记录前一个节点的指针，其实实现起来是很简单的，大家只要看过一次，写过一次，就掌握了。

代码如下：

class Solution {
private:
int result = INT_MAX;
TreeNode* pre = NULL;
void traversal(TreeNode* cur) {
    if (cur == NULL) return;
    traversal(cur->left);   // 左
    if (pre != NULL){       // 中
        result = min(result, cur->val - pre->val);
    }
    pre = cur; // 记录前一个
    traversal(cur->right);  // 右
}
public:
    int getMinimumDifference(TreeNode* root) {
        traversal(root);
        return result;
    }
};

或者大家可以参考之前判断搜索二叉树的博客，我们可以先中序遍历出一个val数组，因为是二叉搜索树，所以这个数组一定是严格递增的，所以后续只需要挨个比较数组相邻元素的差值找出最小即可，代码如下：

class Solution {
public:
    int min = INT_MAX;
    vector<int> nums;
    void tran(TreeNode* node)
    {
        if(node == nullptr) return;
        tran(node->left);
        nums.push_back(node->val);
        tran(node->right);
    }
    int getMinimumDifference(TreeNode* root) {
        tran(root);
        for(int i = 0 ; i < nums.size()-1 ; i++)
        {
            min = min < (nums[i+1] - nums[i]) ? min : (nums[i+1] - nums[i]);
        }
        return min;
    }
};

迭代

看过二叉树的迭代遍历以及统一迭代法之后，不难写出两种中序遍历的迭代法。

下面给出其中的一种中序遍历的迭代法，代码如下：

class Solution {
public:
    int getMinimumDifference(TreeNode* root) {
        stack<TreeNode*> st;
        TreeNode* cur = root;
        TreeNode* pre = NULL;
        int result = INT_MAX;
        while (cur != NULL || !st.empty()) {
            if (cur != NULL) { // 指针来访问节点，访问到最底层
                st.push(cur); // 将访问的节点放进栈
                cur = cur->left;                // 左
            } else {
                cur = st.top();
                st.pop();
                if (pre != NULL) {              // 中
                    result = min(result, cur->val - pre->val);
                }
                pre = cur;
                cur = cur->right;               // 右
            }
        }
        return result;
    }
};