题目描述
给你一个二叉搜索树的根节点 root ,返回 树中任意两不同节点值之间的最小差值 。差值是一个正数,其数值等于两值之差的绝对值。
问题分析
首先要注意二叉树是有序的,可以把二叉搜索树转换为有序数组,然后遍历一遍数组,这里应该中序遍历遍历二叉搜索树,就可以统计最小值了,于是有了递归版本一,然后想可以直接找到相邻两个节点的差值:用pre节点记录cur节点的前一个结点,于是有了递归版本二,用递归版本二的迭代法的方式解决这个问题,总共三种解决方法。
代码实现
// 编程软件:VS2019
// 参考书:代码随想录
#include<iostream>
#include<stack>
#include<vector>
#include<queue>
using namespace std;
// 定义二叉树
struct TreeNode {
int val;
TreeNode* left;
TreeNode* right;
TreeNode(int v):val(v),left(nullptr),right(nullptr){}
};
/*递归版本一
* 二叉搜索树是有序的,把二叉树转换为有序数组,然后遍历一遍数组,就可以统计最小差值
*/
vector<int> vec;
void traversal1(TreeNode* root) {
if (root == NULL) return;
traversal1(root->left);
vec.push_back(root->val);
traversal1(root->right);
}
int getMininumDifference1(TreeNode* root) {
vec.clear();
traversal1(root);
if (vec.size() < 2) return 0;
int result = INT16_MAX;
for (int i = 1; i < vec.size(); i++) { // 统计有序数组的最小差值
result = min(result, vec[i] - vec[i - 1]);
}
return result;
}
/*递归版本二
* 用pre记录cur节点的前一个结点
*/
int result = INT16_MAX;
TreeNode* pre;
void traversal2(TreeNode* cur) {
if (cur == NULL) return;
traversal2(cur->left); // 左
if (pre != NULL) { // 中
result = min(result, cur->val - pre->val);
}
pre = cur; // 记录前一个节点的指针
traversal2(cur->right); // 右
}
int getMininumDifference2(TreeNode* root) {
traversal2(root);
return result;
}
// 迭代法:递归版本二的用迭代的形式表示
int getMininumDifference3(TreeNode* root) {
stack<TreeNode*> st;
TreeNode* cur = root;
TreeNode* pre = NULL;
int result = INT16_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;
}
int main() {
TreeNode* root = new TreeNode(4);
root->left = new TreeNode(2);
root->right = new TreeNode(7);
root->left->left = new TreeNode(1);
root->left->right = new TreeNode(3);
root->right->left = new TreeNode(6);
root->right->right = new TreeNode(8);
cout << getMininumDifference1(root); // 递归版本一
cout << getMininumDifference2(root); // 递归版本二
cout << getMininumDifference3(root); // 迭代法
}