力扣：235. 二叉搜索树的最近公共祖先

题目描述

给定一个二叉搜索树, 找到该树中两个指定节点的最近公共祖先。
最近公共祖先的定义为：“对于有根树 T 的两个结点 p、q，最近公共祖先表示为一个结点 x，满足 x 是 p、q 的祖先且 x 的深度尽可能大（一个节点也可以是它自己的祖先）。”

问题分析

递归法：二叉搜索树是有序的，从上到下遍历过程中，当节点值在[p，q]之间，该节点就是最近公共祖先，所采用先序遍历。
迭代法：利用二叉搜索树的有序性，解决逻辑与递归类似。

代码实现

// 编程软件：VS2019
// 参考书籍：代码随想录
#include<iostream>
using namespace std;

// 定义二叉树
struct TreeNode {
	int val;
	TreeNode* left;
	TreeNode* right;
	TreeNode(int v):val(v),left(nullptr),right(nullptr){}
};

// 递归：二叉搜索树是有序的，从上到下遍历过程中，当节点值在[p，q]之间，该节点就是最近公共祖先，所采用先序遍历
TreeNode* lowestCommonAncestor1(TreeNode* root, TreeNode* p, TreeNode* q) {
	if (root->val > p->val && root->val > q->val) { // 节点值大于p和q就往节点的左孩子遍历
		return lowestCommonAncestor1(root->left, p, q);
	}
	else if (root->val < p->val && root->val < q->val) { // 节点值小于p和q就往节点的右孩子遍历
		return lowestCommonAncestor1(root->right, p, q);
	}
	else return root;
}
// 迭代法：利用二叉搜索树的有序性，解决逻辑与递归类似
TreeNode* lowestCommonAncestor2(TreeNode* root, TreeNode* p, TreeNode* q) {
	while (root) {
		if (root->val > p->val && root->val > q->val) { // 节点值大于p和q就往节点的左孩子遍历
			root = root->left;
		}else if (root->val < p->val && root->val < q->val) { // 节点值小于p和q就往节点的右孩子遍历
			root = root->right;
		}else return root;
	}
	return NULL;
}

int main() {
	TreeNode* root = new TreeNode(6);
	root->left = new TreeNode(2);
	root->right = new TreeNode(8);
	root->left->left = new TreeNode(0);
	root->left->right = new TreeNode(4);
	root->left->right->left = new TreeNode(3);
	root->left->right->right = new TreeNode(5);
	root->right->left = new TreeNode(7);
	root->right->right = new TreeNode(9);
	TreeNode* p = root->left->left; // 节点 0
	TreeNode* q = root->left->right->right; //节点 5
	TreeNode* res = lowestCommonAncestor2(root, p, q);
	cout << res->val << endl;
}
// 结果：2