leetcode 783. 二叉搜索树结点最小距离(Minimum Distance Between BST Nodes)
题目描述:
给定一个二叉搜索树的根结点 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
。 - 二叉树总是有效的,每个节点的值都是整数,且不重复。
解法:
/**
* 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:
void inOrder(TreeNode* root, vector<int>& res){
if(!root){
return;
}
inOrder(root->left, res);
res.push_back(root->val);
inOrder(root->right, res);
}
int minDiffInBST(TreeNode* root) {
vector<int> vals;
inOrder(root, vals);
int sz = vals.size();
int res = vals[1] - vals[0];
for(int i = 2; i < sz; i++){
res = min(vals[i] - vals[i-1], res);
}
return res;
}
};