LeetCode 501 二叉搜索树中的众数 -- 递归法和迭代法

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/find-mode-in-binary-search-tree

---

题意：

给你一个含重复值的二叉搜索树（BST）的根节点 root ，找出并返回 BST 中的所有 众数（即，出现频率最高的元素）。

如果树中有不止一个众数，可以按 任意顺序 返回。

假定 BST 满足如下定义：

结点左子树中所含节点的值 小于等于 当前节点的值
结点右子树中所含节点的值 大于等于 当前节点的值
左子树和右子树都是二叉搜索树

示例 1：

输入：root = [1,null,2,2]
输出：[2]

示例 2：
输入：root = [0]
输出：[0]

提示：
树中节点的数目在范围 [1, 10⁴] 内
-10⁵ <= Node.val <= 10⁵

进阶：你可以不使用额外的空间吗？（假设由递归产生的隐式调用栈的开销不被计算在内）

---

参考文章

思路：

又是一道二叉搜索树的题。涉及到二叉搜索树，要立刻想到，二叉搜索树的中序遍历序列是有序的。

这道题与《LeetCode 98 验证二叉搜索树》和《LeetCode 530 二叉搜索树的最小绝对差》这两道题非常类似，都是利用二叉搜索树的特点，对二叉搜索树进行中序遍历，当遍历到中间节点的时候就进行操作。

以递归法为例，二叉树中序遍历的代码模板是（这里是C++代码，摘自代码随想录）：

void searchBST(TreeNode* cur) {
    if (cur == NULL) return ;
    searchBST(cur->left);       // 左
    （处理节点）                // 中
    searchBST(cur->right);      // 右
    return ;
}

对于中间节点的处理：

这道题中，我们先设一个变量maxCount来记录当前最大的同一个数的出现次数，然后在遍历过程中不断记录遍历到当前的前一个节点，然后将当前节点与前一个节点比较，分为以下情况：
1、如果前一个节点为null，说明当前节点是遍历的第一个节点，将rnCount设置为1
2、如果当前节点与前一个节点值相同，rnCount++
3、如果当前节点与前一个节点值不同，rnCount=1
则得出代码

if (rnNode == null || rnNode.val != node.val) rnCount = 1;
else rnCount++; // rnNode.val == node.val

然后我们再将rnCount与maxCount进行比较：
1、如果rnCount等于maxCount，则将当前节点值加入结果集中
2、如果rnCount大于maxCount，则说明最大次数要改变了，需要将当前结果集清空，然后再将当前节点值加入结果集中
得出代码

if (rnCount == maxCount) ans.add(node.val);
else if (rnCount > maxCount) {
	ans.clear();
	ans.add(node.val);
	maxCount = rnCount;
}

综上，我们分别使用递归法和迭代法来进行中序遍历，以下为代码：

递归法 Java代码：

class Solution {
	TreeNode rnNode;
	int maxCount;
	int rnCount;
	List<Integer> ans;

	private void travel(TreeNode node) {
		if (node == null) return;
		travel(node.left);
		if (rnNode == null || rnNode.val != node.val) rnCount = 1;
		else rnCount++; // rnNode.val == node.val
		if (rnCount == maxCount) ans.add(node.val);
		else if (rnCount > maxCount) {
			ans.clear();
			ans.add(node.val);
			maxCount = rnCount;
		}
		rnNode = node;
		travel(node.right);
	}

	public int[] findMode(TreeNode root) {
		ans = new ArrayList<>();
		travel(root);
		int[] ansArr = new int[ans.size()];
		for (int i = 0; i < ans.size(); i++) {
			ansArr[i] = ans.get(i);
		}
		return ansArr;
	}
}

迭代法 Java代码：

class Solution {
	public int[] findMode(TreeNode root) {
		TreeNode rnNode = null;
		TreeNode curNode = root;
		int maxCount = 0;
		int rnCount = 0;
		Deque<TreeNode> deque = new LinkedList<>();
		List<Integer> ans = new ArrayList<>();
		while (curNode != null || !deque.isEmpty()) {
			if (curNode != null) {
				deque.push(curNode);
				curNode = curNode.left;
			} else {
				curNode = deque.poll();
				if (rnNode == null || curNode.val != rnNode.val) rnCount = 1;
				else rnCount++;
				if (rnCount == maxCount) ans.add(curNode.val);
				else if (rnCount > maxCount) {
					maxCount = rnCount;
					ans.clear();
					ans.add(curNode.val);
				}
				rnNode = curNode;
				curNode = curNode.right;
			}
		}
		int[] ansArr = new int[ans.size()];
		for (int i = 0; i < ans.size(); i++) {
			ansArr[i] = ans.get(i);
		}
		return ansArr;
	}
}