BST两数之和之中序遍历+两指针对撞

二叉树目标和查找

原创于 2019-09-18 08:38:27 发布 · 171 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种在二叉树中寻找两个节点值之和等于特定目标值k的方法。通过中序遍历将二叉树节点值转换为排序数组，然后使用双指针技术在数组中查找匹配项。这是一种高效解决二叉树搜索问题的算法。

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def findTarget(self, root: TreeNode, k: int) -> bool:
        def inorder(root):
            if not root:
                return []
            return inorder(root.left)+[root.val]+inorder(root.right)
        numbers = sorted(inorder(root))
        n = len(numbers)
           
        i = 0
        j = n-1
        while i < j:
             if numbers[i] + numbers[j] > k:
                j -= 1
             elif numbers[i] + numbers[j] < k:
                i +=1
             else:
                return True
        return False