(刷题笔记)Leetcode 653.两数之和 IV - 输入 BST

最新推荐文章于 2022-05-14 23:13:42 发布

原创最新推荐文章于 2022-05-14 23:13:42 发布 · 419 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #二叉树 #c++

刷题笔记专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种在二叉搜索树中寻找两数之和等于特定目标值的有效算法。通过中序遍历得到有序数组，再利用二分查找优化搜索过程，实现了高效的解决方案。

题目

给定一个二叉搜索树和一个目标结果，如果 BST 中存在两个元素且它们的和等于给定的目标结果，则返回 true。

案例 1:

输入:

Target = 9

输出: True

案例 2:

输入:

Target = 28

输出: False

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/two-sum-iv-input-is-a-bst

解题思路

这是一道稍微综合性的题目，考察了二叉树搜索树的性质和二分查找算法。
首先是对二叉搜索树进行中序遍历，用一个vector记录中序遍历结果。
因为二叉搜索树的中序遍历结果是有序的，所以对于K的搜索可以用二分查找法
维护一个left和right的指针（广义的，实际指数组下标），left指向vector的头，right指向vector的尾
比较vec[left]+vec[right]与K的大小，大于K说明vec[right]大了，right–；小于K说明vec[left]小了，left++；
等于直接返回true；跳出条件是left>right的时候，说明没有找到。

话说我一开始写的时候没有想到二分，用双重循环暴力搜索，提交的用时和二分竟然是一样的，不知道是不是测试数据的问题。

代码(C++)

迭代

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool findTarget(TreeNode* root, int k) {

        stack<TreeNode*> st;
        vector<int> vec;
        TreeNode* temp=root;

        while(!st.empty()||temp){

            while(temp){
                st.push(temp);
                temp=temp->left;
            }
            temp=st.top();
            st.pop();
            vec.push_back(temp->val);
            temp=temp->right;
        }

        int left=0;
        int right=vec.size()-1;
        while(left<right){

            if(vec[left]+vec[right]<k) left++;
            else if(vec[left]+vec[right]>k) right--;
            else return true;  
        }

        return false;

    }
};

递归

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    void dfs(TreeNode* root,vector<int> &vec){
        if(root==NULL) return;
        dfs(root->left,vec);
        vec.push_back(root->val);
        dfs(root->right,vec);
    }

    bool findTarget(TreeNode* root, int k) {

        vector<int> vec;
        dfs(root,vec);

        int left=0;
        int right=vec.size()-1;

        while(left<right){
            if(k>vec[left]+vec[right]) left++;
            else if(k<vec[left]+vec[right]) right--;
            else return true;
        }
       
        return false;
    
    }
};