public void recoverTree(TreeNode root) {
Stack<TreeNode> stack = new Stack<>();
TreeNode node = root;
TreeNode prev = null;
TreeNode node1=null,node2=null;
while (!stack.isEmpty()||node!=null){
while (node!=null){
stack.push(node);
node = node.left;
}
node = stack.pop();
if (prev!=null&&node.val<prev.val){
if (node1==null){
node1 = prev;
node2 = node;
}else {
node2 = node;
}
}
prev = node;
node = node.right;
}
if (node1!=null&&node2!=null){
int temp = node1.val;
node1.val = node2.val;
node2.val = temp;
}
}
思路:
题目说到交换节点,那么其实就只有两种可能,交换的节点相邻,或交换的节点不相邻
交换的节点相邻:
ai和ai+1交换,那么就出现了ai>ai+1保存两个节点
这个时候是不是就可以直接交换两个节点了呢?并不是,因为还有一种情况
ai和aj交换,交换后
ai>ai+1,aj-1>aj
发现这两种情况在第一个节点处,情况是一样的那么我们可以先保存第一种情况的两个节点,然后如果出现下一个不符合条件的点时将ai+1用aj替换
然后同一在循环结束后交换两个节点