数据结构之恢复二叉搜索树

(leetcode刷题)

恢复二叉搜索树

   题目: 二叉搜索树中的两个节点被错误地交换。请在不改变其结构的情况下,恢复这棵树。(二叉搜索树是指中序遍历时节点的值按从小到大排序)
   示例: 在这里插入图片描述

方法

  • 我的想法是两边遍历,第一遍按照中序遍历找到所有值,排序,第二遍遍历时改变每个节点的值。
class Solution:     
    def recoverTree(self, root: TreeNode) -> None:        
        """        Do not return anything, modify root in-place instead.        """        			
        if not root:return []        
        stack,ret=[],[]        
        curr=root        
        while stack or curr:            
            if curr:                
            	stack.append(curr)                
            	curr=curr.left            
            else:                
                curr=stack.pop()                
                ret.append(curr.val)                
                curr=curr.right        
        ret.sort(reverse=True)        
        while ret or stack:            
        	if root:                
        	    stack.append(root)                
        	    root=root.left            
        	else:                
        	    root=stack.pop()                
        	    root.val=ret.pop()                
        	    root=root.right

在这里插入图片描述
 

官方方法

在这里插入图片描述
在这里插入图片描述

发布了60 篇原创文章 · 获赞 2 · 访问量 1461

猜你喜欢

转载自blog.csdn.net/qq_40160983/article/details/105054533
今日推荐