leetcode 653. 两数之和 IV - 输入 BST(Two Sum IV - Input is a BST)

最新推荐文章于 2024-03-31 12:09:16 发布

转载最新推荐文章于 2024-03-31 12:09:16 发布 · 52 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zhanzq/p/10605374.html

本文介绍了一种解决二叉搜索树中是否存在两节点之和等于特定目标值的问题的方法。通过中序遍历将节点值存储到数组中，再利用双指针技巧检查是否存在和为目标值的两个数。

题目描述：
示例 1:
示例 2:
解法：

题目描述：

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

示例 1:

输入: 
        5
       / \
      3   6
     / \   \
    2   4   7

Target = 9
输出: True

示例 2:

输入: 
        5
       / \
      3   6
     / \   \
    2   4   7

Target = 28
输出: 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 preOrder(TreeNode* root, vector<int>& nums){
        if(root){
            preOrder(root->left, nums);
            nums.push_back(root->val);
            preOrder(root->right, nums);
        }
    }
    
    bool findTarget(TreeNode* root, int k) {
        vector<int> nums;
        preOrder(root, nums);
        int sz = nums.size();
        int l = 0, r = sz-1;
        while(l < r){
            long long sum_val = nums[l] + nums[r];
            if(sum_val == k){
                return true;
            }else if(sum_val < k){
                l++;
            }else{
                r--;
            }
        }
        return false;
    }
};

转载于:https://www.cnblogs.com/zhanzq/p/10605374.html