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;
}
}
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;
}
}
相关文章推荐
- leetcode[Two Sum IV - Input is a BST]//待整理多种解法
- [LeetCode] Two Sum IV - Input is a BST 两数之和之四 - 输入是二叉搜索树
- LC-Two Sum IV - Input is a BST
- leetcode 653 Two Sum IV - Input is a BST
- LeetCode Two Sum IV - Input is a BST
- LeetCode #653 Two Sum IV - Input is a BST
- leetcode Two Sum IV - Input is a BST 两个数的和
- LeetCode Two Sum IV - Input is a BST
- Two Sum IV - Input is a BST问题及解法
- LeetCode 653: Two Sum IV - Input is a BST
- LeetCode.653 Two Sum IV - Input is a BST
- LeetCode之Two Sum IV - Input is a BST(Kotlin)
- Two Sum IV - Input is a BST
- Two Sum IV - Input is a BST
- 算法分析与设计——LeetCode Problem.653 Two Sum IV - Input is a BST
- Two Sum IV - Input is a BST
- Two Sum IV - Input is a BST(leetcode)
- week19-leetcode #653-TwoSumIV-InputisaBST
- [LeetCode] Two Sum II - Input array is sorted 两数之和之二 - 输入数组有序
- 167.leetcode Two Sum II - Input array is sorted(medium)[两数求和固定值]