两数之和 IV - 输入 BST
给定一个二叉搜索树 root 和一个目标结果 k,如果 BST 中存在两个元素且它们的和等于给定的目标结果,则返回 true。
这题,可以对树进行遍历,然后,在遍历的同时将遍历的数据存储到一个数组中,再从数组中查找k-root->val,如果存在说明有这么两个数,同时设置一个标志rank,如果找到那么也就不需要继续进行遍历了,可以返回true。
代码如下:
void deepsearch(struct TreeNode* root, int* a,int *size,int k,int *rank){
int i;
if(root!=NULL&&*(rank)==0){
a[*size]=root->val;
(*size)++;
for(i=0;i<(*size)-1;i++){
if(a[i]==k-(root->val)) (*rank)++;
}
deepsearch(root->left,a,size,k,rank);
deepsearch(root->right,a,size,k,rank);
}
}
bool findTarget(struct TreeNode* root, int k){
int a[10000],size=0;
int i,j;
int rank=0;
deepsearch(root,a,&size,k,&rank);
if(rank!=0)
return true;
else return false;
}