打印总和大于给定值的两个 BST 中的所有对

介绍

这篇文章会介绍如何解决一个问题：如何找到两个二叉搜索树，使得它们节点之和大于给定值。我们将使用语言Java来实现。

思路

这个问题看起来很简单，我们可以先中序遍历两个二叉搜索树并将它们的节点值存储到两个数组中，然后在这两个数组中寻找节点值之和大于给定值的所有节点对。这个思路虽然可行，但是时间复杂度为O(n)，不是很好的解法。

一个更好的解法是使用“双指针”，这个技巧通常用于对两个有序数组进行操作。我们可以在第一个二叉搜索树中打一个指针指向最小值，再在第二个二叉搜索树中打一个指针指向最大值，然后根据两个节点的值之和与给定值的大小关系来调整指针的位置。

代码

public void findPairs(TreeNode root1, TreeNode root2, int target) {
    List<Integer> list1 = new ArrayList<>();
    List<Integer> list2 = new ArrayList<>();
    inorder(root1, list1);
    inorder(root2, list2);

    int i = 0, j = list2.size() - 1;
    while (i < list1.size() && j >= 0) {
        int sum = list1.get(i) + list2.get(j);
        if (sum > target) {
            j--;
        } else {
            i++;
        }
    }
}

// 中序遍历将节点值存储到数组中
private void inorder(TreeNode root, List<Integer> list) {
    if (root == null) {
        return;
    }
    inorder(root.left, list);
    list.add(root.val);
    inorder(root.right, list);
}

上面的代码中，我们使用了一个inorder方法来实现中序遍历，将节点值存储到数组中。然后我们使用两个指针i和j来遍历两个数组，直到找到所有的节点对。

结论

本文介绍了如何找到两个二叉搜索树，使得它们节点之和大于给定值，并给出了一个更好的解法，即使用“双指针”技巧。此解法的时间复杂度为O(n)，是一种较好的解决方案。