Leetcode——653. Two Sum IV - Input is a BST

题目描述

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

解题思路

给定一个BSF二叉树,判断二叉树中是否有两个节点的值之和等于目标值,如果有则返回true,否则返回false

这个题有很多种解法,我采取的方法是首先将二叉树中的元素遍历到一个Set集合中,然后判断集合中是否有两个符合题意的节点。遍历的方法使用中序遍历的方法,借助stack;判断集合是否有符合题意的两个节点采用迭代器。

AC代码

class Solution {
    public boolean findTarget(TreeNode root, int k) {
        Set<Integer> set = new HashSet<>();
        Stack<TreeNode> stack = new Stack<>();
        while(stack.size() > 0 || root != null) {
            if(root != null) {
                stack.push(root);
                root = root.left;
            }else {
                root = stack.pop();
                set.add(root.val);
                root = root.right;
            }
        }
        Iterator<Integer> i = set.iterator();
        while(i.hasNext()) {
            int target = i.next();
            if(set.contains(k - target) && (k - target) != target)
                return true;
        }
        return false;        
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值