leetcode653. Two Sum IV - Input is a BST

BEconfidence

于 2017-08-30 23:42:16 发布

阅读量238

点赞数

CC 4.0 BY-SA版权

分类专栏：每日一题

本文链接：https://blog.youkuaiyun.com/a197p/article/details/77727335

每日一题专栏收录该内容

60 篇文章

订阅专栏

本文介绍了一种方法来判断二叉树中是否存在两个节点，它们的值相加等于特定的目标值k。通过遍历整个二叉树并收集每个节点的值到一个数组中，再检查数组中是否存在两个不同的索引位置，使得这两个位置对应的值之和等于目标值k。

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

题意是求二叉树中的任意两个数的和能不能是k。

把二叉树便利一遍，求出所有元素，然后在循环便利每两个的和就好。

/**
 * 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:
    vector<int> s;
    void f(TreeNode* root){
        if(root!=NULL) s.push_back(root->val);
        if(root->left!=NULL)
        f(root->left);
         if(root->right!=NULL)
        f(root->right);
    }
    bool findTarget(TreeNode* root, int k) {
        f(root);
        int n=s.size();
        for(int i=0;i<n;i++){
            for(int j=i+1;j<n;j++){
                if(s[i]+s[j]==k) return 1;
            }
        }
        return 0;
    }
};