标签:pre JAVA val st 99 firstNode null root LeetCode
二叉搜索树中的两个节点被错误地交换。
请在不改变其结构的情况下,恢复这棵树。
示例 1:
输入: [1,3,null,null,2]
1
/
3
\
2
输出: [3,1,null,null,2]
3
/
1
\
2
示例 2:
输入: [3,1,4,null,null,2]
3
/ \
1 4
/
2
输出: [2,1,4,null,null,3]
2
/ \
1 4
/
3
思路:我们可以考虑到上一题中的方法三:验证二叉搜索树,采用中序遍历的方法。因为我们知道,一个合格的二叉搜索树在中序遍历后得到的序列一定是一个升序序列,当有两个点被交换时一定会出现两次前一个点或者一次(相邻节点被交换了)前一个点比后一个点大的情况,因此我们找到这两个点,然后保存一下,之后交换回来即可, 因为只用交换值,所以这题中规中矩吧。
class Solution {
public void recoverTree(TreeNode root) {
Stack<TreeNode> st=new Stack<TreeNode>();
TreeNode firstNode=null,secondNode=null;
TreeNode pre=new TreeNode(Integer.MIN_VALUE);
while(!st.isEmpty() || root!=null)
{
while(root!=null)
{
st.push(root);
root=root.left;
}
root=st.pop();
if(firstNode==null && pre.val>=root.val) firstNode=pre;
if(firstNode!=null && pre.val>=root.val) secondNode=root;
pre=root;
root=root.right;
}
int tmp=firstNode.val;
firstNode.val=secondNode.val;
secondNode.val=tmp;
}
}
标签:pre,JAVA,val,st,99,firstNode,null,root,LeetCode 来源: https://blog.csdn.net/haut_ykc/article/details/99286640
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。