653. Two Sum IV - Input is a BST

最新推荐文章于 2024-06-02 16:09:57 发布

原创最新推荐文章于 2024-06-02 16:09:57 发布 · 277 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

429 篇文章

订阅专栏

杭code

7 篇文章

订阅专栏

本文介绍了一种通过将二叉搜索树转换为数组的方法来高效查找两个节点值之和是否等于特定目标值的算法。该算法首先使用深度优先搜索遍历二叉树并收集节点值到一个数组中，然后利用双指针技术在已排序的数组中寻找符合条件的节点值组合。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

先转换为一个数组序列，然后判断是否存在两个数字，变成了一道简单题

/**
 * 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>& nums)
    {
        if(root==NULL)
            return;

        dfs(root->left,nums);
        nums.push_back(root->val);
        dfs(root->right, nums);

    }

    bool findTarget(TreeNode* root, int k) {

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

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

        while(left<right){
            if(nums[left]+nums[right]==k)
                return true;
            else if(nums[left]+nums[right]<k)
                left++;
            else
                right--;
        }

        return false;
    }
};