数据结构
网络
关系数据库管理系统 (RDBMS)
操作系统
Java
Microsoft Excel
iOS
HTML
CSS
Android
Python
C语言编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理学
化学
生物学
数学
英语
经济学
心理学
社会学
服装设计
法学C++实现两数之和 IV - 输入为二叉搜索树
假设我们有一个二叉搜索树和一个目标值；我们需要检查二叉搜索树中是否存在两个元素，它们的和等于给定的目标值。
因此，如果输入如下所示：
则输出为True。
为了解决这个问题，我们将遵循以下步骤：
定义一个数组v
定义一个函数inorder()，它将接收根节点作为参数：
如果根节点为空，则：
返回
inorder(根节点的左子树)
将根节点的值插入到v中
inorder(根节点的左子树)
inorder(根节点的右子树)
定义一个函数findnode()，它将接收k作为参数：
n := v的大小
当 i < j 时，执行以下操作：
t := v[i] + v[j]
如果t等于k，则：
返回true
如果t < k，则：
（将i增加1）
否则
（将j减少1）
返回false
在主方法中执行以下操作：
inorder(root)
对数组v进行排序
返回findnode(k)
示例
#include <bits/stdc++.h>
using namespace std;
class TreeNode{
public:
int val;
TreeNode *left, *right;
TreeNode(int data){
val = data;
left = NULL;
right = NULL;
}
};
void insert(TreeNode **root, int val){
queue<TreeNode*> q;
q.push(*root);
while(q.size()){
TreeNode *temp = q.front();
q.pop();
if(!temp->left){
if(val != NULL)
temp->left = new TreeNode(val);
else
temp->left = new TreeNode(0);
return;
}
else{
q.push(temp->left);
}
if(!temp->right){
if(val != NULL)
temp->right = new TreeNode(val);
else
temp->right = new TreeNode(0);
return;
}
else{
q.push(temp->right);
}
}
}
TreeNode *make_tree(vector<int> v){
TreeNode *root = new TreeNode(v[0]);
for(int i = 1; i<v.size(); i++){
insert(&root, v[i]);
}
return root;
}
class Solution {
public:
vector<int> v;
void inorder(TreeNode* root){
if (root == NULL || root->val == 0)
return;
inorder(root->left);
v.push_back(root->val);
inorder(root->right);
}
bool findnode(int k){
int n = v.size(), i = 0, j = n - 1;
while (i < j) {
int t = v[i] + v[j];
if (t == k)
return true;
if (t < k)
i++;
else
j--;
}
return false;
}
bool findTarget(TreeNode* root, int k){
inorder(root);
sort(v.begin(), v.end());
return findnode(k);
}
};
main(){
Solution ob;
vector<int> v = {5,3,6,2,4,NULL,7};
TreeNode *root = make_tree(v);
cout << (ob.findTarget(root, 9));
}在线演示
{5,3,6,2,4,NULL,7},9输入
1
Arnab Chakraborty
更新于：2020年6月11日
打印页面