您的位置:首页 > 其它

Two Sum IV - Input is a BST

2017-10-30 15:26 369 查看
题目描述:

Given a Binary Search Tree and a target number, return true if there exist two elements in the BST such that their sum is equal to the given target.Example 1:
Input:
5
/ \
3 6
/ \ \
2 4 7

Target = 9

Output: True
Example 2:
Input:
5
/ \
3 6
/ \ \
2 4 7

Target = 28

Output: False
分析:直接在二叉树上进行搜索比较困难,可以先采用前序遍历将二叉树转换成有序数组,然后采用首位指针的方式搜索结果。代码:/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public boolean findTarget(TreeNode root, int k) {
if(root == null){
return false;
}

List<Integer> result = BST2Array(root);
if(result == null || result.size() <= 1){
return false;
}
// 首尾遍历有序数组
int i = 0;
int j = result.size()-1;
while (i < j){
int sum = result.get(i) + result.get(j);
if(sum == k){
return true;
}
if(sum < k){
i++;
}else {
j--;
}

}
return false;
}

private List<Integer> BST2Array(TreeNode root) {
if(root == null) {
return null;
}
// 前序遍历
List<Integer> result = new ArrayList<>();
List<Integer> left = BST2Array(root.left);
List<Integer> right = BST2Array(root.right);

if(left != null){
result.addAll(left);
}
result.add(root.val);

if(right!= null){
result.addAll(right);
}
return result;
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: