难度中等705收藏分享切换为英文接收动态反馈
给你二叉搜索树的根节点 root
,该树中的 恰好 两个节点的值被错误地交换。请在不改变其结构的情况下,恢复这棵树 。
示例 1:
输入:root = [1,3,null,null,2] 输出:[3,1,null,null,2] 解释:3 不能是 1 的左孩子,因为 3 > 1 。交换 1 和 3 使二叉搜索树有效。
示例 2:
输入:root = [3,1,4,null,null,2] 输出:[2,1,4,null,null,3] 解释:2 不能在 3 的右子树中,因为 2 < 3 。交换 2 和 3 使二叉搜索树有效。
提示:
- 树上节点的数目在范围
[2, 1000]
内 -231 <= Node.val <= 231 - 1
进阶:使用 O(n)
空间复杂度的解法很容易实现。你能想出一个只使用 O(1)
空间的解决方案吗?
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
//左根右,正常依次增大,异常是根大于左,小于右。找第一个时,lastElement相当于父根,大于root是异常。找第二个时lastElement相当于孩子,root是父节点,root<lastElement
class Solution {
private TreeNode firstElement = null;
private TreeNode secondElement = null;
private TreeNode lastElement = new TreeNode(Integer.MIN_VALUE);
private void traverse(TreeNode root) {
if (root == null) {
return;
}
traverse(root.left);
if (firstElement == null && root.val < lastElement.val) {
firstElement = lastElement;
}
if (firstElement != null && root.val < lastElement.val) {
secondElement = root;
}
lastElement = root;
traverse(root.right);
}
public void recoverTree(TreeNode root) {
// traverse and get two elements
traverse(root);
// swap
int temp = firstElement.val;
firstElement.val = secondElement.val;
secondElement.val = temp;
}
}