题目地址：

https://leetcode.com/problems/two-sum-bsts/

给定两棵二叉搜索树，再给定一个数xxx，问两棵BST里是否各自存在一个节点使得它们之和恰好等于xxx。

在两个序列上做这件事是容易的，可以直接用对撞双指针（一个指针指向第一个数组的开头，另一个指向第二个数组的结尾）。现在要在两棵BST上做，思路是一样的，可以先实现二叉树的中序遍历的迭代器，以及逆序中序遍历的迭代器，然后就可以当成数组的情况来做了。

中序遍历迭代器的写法是用栈，初始化的时候，先将树根所在的左链push进栈，每次弹出的节点即为下一个走到的节点，走到之后，将其左儿子所在的左链push进栈（如果有的话）。逆序中序遍历迭代器的写法和正序中序遍历迭代器的写法是对称的。

代码如下：

class Solution {public:struct Ite {stack<TreeNode *> stk;bool desc;Ite(TreeNode *root, bool desc) : desc(desc) {while (root) {stk.push(root);root = !desc ? root->left : root->right;}}// 重载bool()运算符operator bool() { return stk.size(); }// 重载dereference运算符int operator*() { return stk.top()->val; }// 重载前置++运算符Ite &operator++() {auto t = stk.top(); stk.pop();if (!desc) {auto right = t->right;while (right) {stk.push(right);right = right->left;}} else {auto left = t->left;while (left) {stk.push(left);left = left->right;}}return *this;}};bool twoSumBSTs(TreeNode *r1, TreeNode *r2, int t) {Ite it1(r1, false), it2(r2, true);while (it1 && it2) {int x1 = *it1, x2 = *it2;if (x1 + x2 < t) ++it1;else if (x1 + x2 > t) ++it2;else return true;}return false;}
};

时间复杂度O(n1+n2)O(n_1+n_2)O(n1+n2)，空间O(h1+h2)O(h_1+h_2)O(h1+h2)。

【Leetcode】1214. Two Sum BSTs相关推荐

【LeetCode】1. Two Sum
传送门:https://leetcode.com/problems/two-sum/#/description 一.题目描述 Given an array of integers, return in ...
【leetcode】Minimum Path Sum
Minimum Path Sum Given a m x n grid filled with non-negative numbers, find a path from top left to b ...
【LeetCode】#39组合总和(Combination Sum)
[LeetCode]#39组合总和(Combination Sum) 加粗样式给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数 ...
【LeetCode】【HOT 100】2. 两数相加
[LeetCode][HOT 100]2. 两数相加文章目录 [LeetCode][HOT 100]2. 两数相加 package hot;import java.util.Arrays;class ...
【LeetCode】剑指 Offer 34. 二叉树中和为某一值的路径
[LeetCode]剑指 Offer 34. 二叉树中和为某一值的路径文章目录 [LeetCode]剑指 Offer 34. 二叉树中和为某一值的路径 package offer;import ja ...
【LeetCode】剑指 Offer 13. 机器人的运动范围
[LeetCode]剑指 Offer 13. 机器人的运动范围文章目录 [LeetCode]剑指 Offer 13. 机器人的运动范围 package offer;public class Solu ...
【LeetCode】第643题——子数组最大平均数I（难度：简单）
[LeetCode]第643题--子数组最大平均数I(难度:简单) 题目描述解题思路代码详解注意点题目描述给定 n 个整数,找出平均数最大且长度为 k 的连续子数组,并输出该最大平均数. 示 ...
【leetcode】771. 宝石与石头（简单）
[leetcode]771. 宝石与石头给你一个字符串 jewels 代表石头中宝石的类型,另有一个字符串 stones 代表你拥有的石头. stones 中每个字符代表了一种你拥有的石头的类型,你 ...
【LeetCode】1641. Count Sorted Vowel Strings（动态规划）
[LeetCode]1641. Count Sorted Vowel Strings(动态规划) Given an integer n, return the number of strings of ...
【leetcode】二维子矩阵的和
一. 题目描述给定一个二维矩阵 matrix,以下类型的多个请求: 计算其子矩形范围内元素的总和,该子矩阵的左上角为 : (row1, col1) ,右下角为 (row2, col2) . 实现 Nu ...

【Leetcode】1214. Two Sum BSTs

题目地址：

【Leetcode】1214. Two Sum BSTs相关推荐

最新文章

热门文章