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

目录

• 题目
• 解题思路
• 代码(C++)

题目

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

案例 1:

输入:

5
   / \
  3   6
 / \   \
2   4   7
           

Target = 9

输出: True

案例 2:

输入:

5
   / \
  3   6
 / \   \
2   4   7
           

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;
    
    }
};