LeetCode 99. 恢复二叉搜索树 python

99. 恢复二叉搜索树

二叉搜索树中的两个节点被错误地交换。

请在不改变其结构的情况下，恢复这棵树。

示例 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

方法一：显式中序遍历:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def findTwoSwapped(self, nums):
        x, y = 0, 0
        for i in range(len(nums) - 1):
            if nums[i] > nums[i + 1]:
                if x == 0:
                    x = nums[i]
                y = nums[i + 1]
        return x, y
    def recover(self, root