前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >LeetCode 1214. 查找两棵二叉搜索树之和(二叉树迭代器+双指针)

LeetCode 1214. 查找两棵二叉搜索树之和(二叉树迭代器+双指针)

作者头像
Michael阿明
发布2020-07-13 16:33:16
1.2K0
发布2020-07-13 16:33:16
举报
文章被收录于专栏:Michael阿明学习之路

1. 题目

给出两棵二叉搜索树,请你从两棵树中各找出一个节点,使得这两个节点的值之和等于目标值 Target。

如果可以找到返回 True,否则返回 False。

代码语言:javascript
复制
示例 1:
输入:root1 = [2,1,4], root2 = [1,0,3], target = 5
输出:true
解释:2 加 3 和为 5 。

示例 2:
输入:root1 = [0,-10,10], root2 = [5,1,7,0,2], target = 18
输出:false
 
提示:
每棵树上最多有 5000 个节点。
-10^9 <= target, node.val <= 10^9

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/two-sum-bsts 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2. 解题

  • 写一个二叉搜索树的迭代器,一个升序,一个降序
  • 双指针一个指向最小的,一个指向最大的,根据情况决定,哪个往下挪
代码语言:javascript
复制
class bst_iter
{
	stack<TreeNode*> s;
	TreeNode* cur;
	int v;
	bool postorder;
public:
	bst_iter(TreeNode* root, bool postorder = false)
	{
		cur = root;
		this->postorder = postorder;//降序
	}
	int next()
	{
		if(!postorder && (cur || !s.empty()))
		{
			while(cur)
			{
				s.push(cur);
				cur = cur->left;
			}
			v = s.top()->val;
			cur = s.top()->right;
			s.pop();
			return v;
		}
		else if(postorder && (cur || !s.empty()))
		{
			while(cur)
			{
				s.push(cur);
				cur = cur->right;//降序先遍历右边
			}
			v = s.top()->val;
			cur = s.top()->left;
			s.pop();
			return v;
		}
		return INT_MAX;
	}
};
class Solution {
public:
    bool twoSumBSTs(TreeNode* root1, TreeNode* root2, int target) {
    	bst_iter bstit1(root1);//ascending
    	bst_iter bstit2(root2,true);//descending
    	int v1 = bstit1.next(), v2 = bstit2.next();
    	while(v1 != INT_MAX && v2 != INT_MAX)
    	{
    		if(v1+v2 < target)
    			v1 = bstit1.next();
    		else if(v1+v2 > target)
    			v2 = bstit2.next();
    		else
    			return true;
    	}
    	return false;
    }
};

48 ms 28.7 MB

本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2020/07/04 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 1. 题目
  • 2. 解题
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档