C++刷Leetcode 783.二叉搜索树结点最小距离

给定一个二叉搜索树的根结点 root, 返回树中任意两节点的差的最小值。

示例:

输入: root = [4,2,6,1,3,null,null]
输出: 1
解释:
注意,root是树结点对象(TreeNode object),而不是数组。

给定的树 [4,2,6,1,3,null,null] 可表示为下图:

          4
        /   \
      2      6
     / \    
    1   3  

最小的差值是 1, 它是节点1和节点2的差值, 也是节点3和节点2的差值。
注意:

二叉树的大小范围在 2 到 100。
二叉树总是有效的,每个节点的值都是整数,且不重复。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/minimum-distance-between-bst-nodes

 思路

二叉搜索树的中序遍历即为从小到大的有序数组
任意两个结点的距离一定大于等于相邻两个结点的距离

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int minDiffInBST(TreeNode* root) {
        int min = INT_MAX;
        stack s;
        TreeNode *p = root, *pre = nullptr;
        while(!s.empty() || p)
        {
            while(p)
            {
                s.push(p);
                p = p -> left;
            }
            if(!s.empty())
            {
                if(pre)
                {
                    if(s.top() -> val - pre -> val < min)
                        min = s.top() -> val - pre -> val;
                }
                pre = s.top();
                p = pre -> right;
                s.pop();
            }
        }
        return min;
    }
};

 

你可能感兴趣的:(C++,Leetcode,二叉搜索树,Leetcode,最小距离)