Two elements of a binary search tree (BST) are swapped by mistake.
Recover the tree without changing its structure.
Note:
A solution using O(n) space is pretty straight forward. Could you devise a constant space solution?
一刷
題解:
我們用中序遍歷简肴,這樣子如果是正常順序應該是順序的辩撑。
碰到第一個不滿足要求的點,相鄰的點左(較大)>右,那么第一個不滿足要求的為左邊這個點。碰到第二個不滿足要求的點,左>右(較小)宁仔。因為之后要交換這兩個點。那么第二個不滿足要求的為右邊的點峦睡。然后直接交換兩個節(jié)點的val就可以翎苫。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
TreeNode firstNode = null;
TreeNode secondNode = null;
TreeNode prev = new TreeNode(Integer.MIN_VALUE);
public void recoverTree(TreeNode root) {
//find the node
traverse(root);
//swap
int val = firstNode.val;
firstNode.val = secondNode.val;
secondNode.val = val;
}
private void traverse(TreeNode root){
if(root==null) return;
traverse(root.left);
if(firstNode==null && prev.val > root.val){
firstNode = prev;
}
if(firstNode!=null && prev.val > root.val){
secondNode = root;
}
prev = root;
traverse(root.right);
}
}
二刷
思路同一刷
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
TreeNode first = null;
TreeNode second = null;
TreeNode prev = new TreeNode(Integer.MIN_VALUE);
public void recoverTree(TreeNode root) {
traverse(root);
//swap
int temp = first.val;
first.val = second.val;
second.val = temp;
}
private void traverse(TreeNode root){
if(root == null) return;
traverse(root.left);
//swap the prev and root
if(first == null && prev.val>root.val) first = prev;
if(first!=null && prev.val>root.val) second = root;
prev = root;
traverse(root.right);
}
}