二叉搜索树 中序遍历 查找两数之和

给定一个二叉搜索树和一个目标结果,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。
在这里插入图片描述

思路

1.二叉搜索树的中序遍历是递增序列,反中序遍历(先找右节点)是递减数组
2.在有序数组中找两数之和为k时,可以同时比较最大最小值
大于k,从右端找更小值;
小于k,从左端找更大值;
代码结合以上两点:

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    
    bool findTarget(TreeNode* root, int k) {
        TreeNode* cur_left=root;
        TreeNode* cur_right=root;
        stack<TreeNode*> st_left;  //存放中序遍历节点,从小到大
        stack<TreeNode*> st_right; //存放反中序遍历节点 ,从大到小
        int l,r;
        while(cur_left||cur_right||!st_left.empty()||!st_right.empty())
        {
            if(cur_left)
            {
                st_left.push(cur_left);//记录中序节点
                r=cur_left->val;
                cur_left=cur_left->left;
            }
            if(cur_right)
            {
                st_right.push(cur_right);
                l=cur_right->val;
                cur_right=cur_right->right;
                if(l+r==k&&l!=r)  //遍历同时进行判断
                    return true;
            } 
            if(!cur_right&&!cur_left)
            {
                cur_left=st_left.top();//中序遍历最左边节点,必定是最小节点
                cur_right=st_right.top(); //反中序遍历最右边节点,必定是最大节点
                if(cur_left==cur_right)
                    break;
                l=cur_left->val;
                r=cur_right->val;
                if(l+r==k)
                    return true;
                else if(l+r>k)
                {
                    //需要更小值,左边节点不变,右边进行反中序遍历,找更小的值
                    cur_left=NULL;
                    cur_right=cur_right->left;
                    st_right.pop();
                }
                else
                {
                    //需要更大值,右边节点不变,左边进行中序遍历,找更大的值
                    cur_right=NULL;
                    cur_left=cur_left->right;
                    st_left.pop();
                }
            }
        }  
        return false;

    }
};


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值