两数之和 - BST版本-LintCode

最新推荐文章于 2021-11-30 00:18:25 发布

zhaokane

最新推荐文章于 2021-11-30 00:18:25 发布

阅读量288

点赞数

分类专栏： LintCode

本文链接：https://blog.csdn.net/zhaohengchuan/article/details/78589018

版权

LintCode 专栏收录该内容

428 篇文章 2 订阅

订阅专栏

给一棵二叉搜索树以及一个整数 n, 在树中找到和为 n 的两个数字

注意事项：
Without any extra space.

样例：
给一棵BST:

以及一个整数 n = 3
返回 [1, 2] 或 [2, 1]

思路：
遍历节点，求得节点值与n的差，从根开始寻找值为差的节点是否存在。

#ifndef C689_H
#define C689_H
#include<iostream>
#include<vector>
using namespace std;
class TreeNode{
public:
    int val;
    TreeNode *left, *right;
    TreeNode(int val)
    {
        this->val = val;
        this->left = this->right = NULL;
    }
};
class Solution {
public:
    /*
    * @param : the root of tree
    * @param : the target sum
    * @return: two number from tree witch sum is n
    */
    vector<int> twoSum(TreeNode * root, int n) {
        // write your code here
        vector<int> res;
        if (root == NULL)
            return res;
        TreeNode *p = root;
        createVec(root, p, n, res);
        return res;
    }
    //p为root中的任意一个节点,找到满足条件的节点则返回,否则继续遍历
    void createVec(TreeNode *root, TreeNode *p, int n,vector<int> &res)
    {
        if (p == NULL)
            return;
        if (isValid(root, n - p->val))
        {
            res.push_back(p->val);
            res.push_back(n - p->val);
            return;
        }
        else
        {
            createVec(root, p->left, n, res);
            createVec(root, p->right, n, res);
        }
    }
    //判断以root为根的树中,是否存在值为n的节点
    bool isValid(TreeNode *root, int n)
    {
        if (root == NULL)
            return false;
        if (n > root->val)
        {
            if (root->right == NULL)
                return false;
            else
                return isValid(root->right, n);
        }
        else if (n == root->val)
            return true;
        else
        {
            if (root->left == NULL)
                return false;
            else
                return isValid(root->left, n);
        }
    }
};
#endif