[LeetCode]99. 恢复二叉搜索树(java实现)中序遍历找逆序对
1. 题目
2. 读题(需要重点注意的东西)
由于二叉搜索树的中序遍历是有序的,因此找到逆序对,即可判断出哪两个节点进行了交换,逆序对有如下两种情况:
-
只有一个逆序对:则这两个节点就是进行交换的节点
-
有两个逆序对:则第一个逆序对的第一个节点和第二个逆序对的第二个节点就是进行交换的两个节点
思路(中序遍历找逆序对):
递归找到二叉搜索树中序遍历中的逆序对,然后进行交换即可,空间复杂度为O(n)。
不需要开一个数组存储中序遍历的值,只需要有一个节点记录上一个出现的节点即可,当上一个节点的值大于当前节点的值,说明出现了逆序。
3. 解法
---------------------------------------------------解法---------------------------------------------------
/**
* 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;
* }
* }
*/
class Solution {
TreeNode l ,r ,pre;
public void recoverTree(TreeNode root) {
inorder(root);
int temp = l.val;
l.val = r.val;
r.val = temp;
}
private void inorder(TreeNode root){
if(root == null) return;
inorder(root.left);
// 如果上一个节点的值大于当前节点的值,说明有逆序
if(pre != null && pre.val > root.val){
// 不管哪种情况,逆序的值都是第一个出现的逆序数和最后一个出现的逆序数
// 如果逆序数 l 是第一次出现,则赋值
if(l == null) l = pre;
// r为最后一个出现的逆序数
r = root;
}
pre = root;
inorder(root.right);
}
}
可能存在的问题: