博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
两数之和(输入为二叉树) Two Sum IV - Input is a BST
阅读量:6096 次
发布时间:2019-06-20

本文共 1713 字,大约阅读时间需要 5 分钟。

  hot3.png

问题:

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   7Target = 9Output: True

Example 2:

Input:     5   / \  3   6 / \   \2   4   7Target = 28Output: False

解决:

① 中序遍历二叉搜索树可以得到一个递增数组,只要查找该数组中是否包含两个数之和为target。

/**

 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
public class Solution { // 33ms
    public boolean findTarget(TreeNode root, int k) {
        if(root == null) return false;
        List<Integer> list = new ArrayList<>();
        inorder(root,list);
        int head = 0;
        int tail = list.size() - 1;
        while(head < tail){
            if(list.get(head) + list.get(tail) == k){
                return true;
            }else if(list.get(head) + list.get(tail) < k){
                head ++;
            }else{
                tail --;
            }
        }
        return false;
    }
    public void inorder(TreeNode node,List<Integer> list){
        if(node == null) return;
        inorder(node.left,list);
        list.add(node.val);
        inorder(node.right,list);
    }
}

② 先序遍历二叉搜索树,根据遍历到的节点再查找target- node.val即可。

public class Solution { //23ms

    public boolean findTarget(TreeNode root, int k) {
        return helper(root, root, k);
    }
    public boolean helper(TreeNode root, TreeNode curNode, int k) {
        if (curNode == null) return false;
        return preorder(root, curNode, k - curNode.val) || helper(root, curNode.left, k) || helper(root, curNode.right, k);
    }
    public boolean preorder(TreeNode root, TreeNode curNode, int k) {
        if (root == null) return false;
        if (root != curNode && root.val == k) return true;
        return (root.val < k) ? preorder(root.right, curNode, k) : preorder(root.left, curNode, k);
    }
}

转载于:https://my.oschina.net/liyurong/blog/1506331

你可能感兴趣的文章
linux 中文件类型和颜色的区分
查看>>
cocosPods 常见使用步骤
查看>>
对你同样重要的非技术贴,8个方法让你的老板认可你
查看>>
MLP、RBF、SVM神经网络比较
查看>>
最常用的命令
查看>>
mysql数据库备份小记录
查看>>
WordPress 手机客户端生成系统 NextApp 配置指南
查看>>
字典 dict
查看>>
iOS 9 sdk bitcode
查看>>
windows上一键自动安装zabbix-agent
查看>>
SFB 项目经验-50-Lync-And-Cisco-功能-产品-对比
查看>>
iOS OC 异常处理
查看>>
树形菜单文档 - layui.tree-示例
查看>>
array_filter
查看>>
我的友情链接
查看>>
Swift中的方法(Methods)
查看>>
19个Linux备份压缩命令
查看>>
MySQL 报错 ‘SSL_OP_NO_COMPRESSION’ 未声明 (在此函数内第一次使用)
查看>>
scp报错:not a regular file
查看>>
xml中定义个TextView控件及java代码中调用方法。
查看>>