LeetCode算法题-Two Sum IV - Input is a BST(Java实现)

这是悦乐书的第280次更新,第296篇原创

01 看题和准备

今天介绍的是LeetCode算法题中Easy级别的第148题(顺位题号是653)。给定二进制搜索树和目标数,如果BST中存在两个元素,使得它们的总和等于给定目标,则返回true。例如:

    5
   / \
  3   6
 / \   \
2   4   7

目标值:9

输出:true

    5
   / \
  3   6
 / \   \
2   4   7

目标值:28

输出:false

本次解题使用的开发工具是eclipse,jdk使用的版本是1.8,环境是win7 64位系统,使用Java语言编写和测试。

02 第一种解法

判断在给出的一组数据中,是否存在两个数之和等于给出的目标值,只是给出的数据是个二叉搜索树。所以,我们需要先将数据抽出来,遍历二叉树的节点值,然后再去其中找那可能存在的两个数。

/**
 * 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) {
        Map<Integer, Integer> map = new HashMap<Integer,Integer>();
        preOrder(root, map);
        // 遍历map
        for (Integer key : map.keySet()) {
            // 分两种情况:k由两个相同的节点值之和组成;由不同的节点值之和组成
            if ((k-key == key && map.get(key) >= 2) || (k-key != key && map.containsKey(k-key))) {
                return true;
            }
        }        
        return false;
    }

    // 使用递归,前序遍历,将二叉树的节点值添加进map中
    public void preOrder(TreeNode node, Map<Integer, Integer> map) {
        if (node == null) {
            return ;
        }
        map.put(node.val, map.getOrDefault(node.val, 0)+1);
        preOrder(node.left, map);
        preOrder(node.right, map);
    }
}


03 第二种解法

我们也可以不使用HashMap,直接使用数组,另外,题目还说了此二叉树是个BST,那么它的中序遍历的节点值是有序的,那么我们可以直接使用双指针而不必排序,对所有的节点值直接判断。

/**
 * 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) {
        List<Integer> list = new ArrayList<Integer>();
        inOrder(root, list);
        // 使用双指针
        int start = 0, end = list.size()-1;
        while (start < end) {
            int sum = list.get(start) + list.get(end);
            if (sum == k) {
                return true;
            } 
            if (sum > k) {
                end--;
            } else {
                start++;
            }
        }
        return false;
    }

    // 使用递归,中序遍历,将二叉树的节点值添加进list中
    public void inOrder(TreeNode node, List<Integer> list) {
        if (node == null) {
            return ;
        }
        // 左子树
        inOrder(node.left, list);
        // 根节点
        list.add(node.val);
        // 右子树
        inOrder(node.right, list);
    }
}


04 第三种解法

我们也可以使用HashSet,思路和第一种解法类似,依旧使用递归遍历BST的节点,但是此解法是直接在递归方法里面做了判断,如果里面存在两个节点值的话,就直接返回true。

/**
 * 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) {
        Set<Integer> set = new HashSet<Integer>();
        return helper(root, set, k);
    }

    // 使用递归,将二叉树的节点值添加进set中
    public boolean helper(TreeNode node, Set<Integer> set, int k) {
        if (node == null) {
            return false;
        }
        if (set.contains(k-node.val)) {
            return true;
        }
        set.add(node.val);
        return helper(node.left, set, k) || helper(node.right, set, k);
    }
}


05 第四种解法

针对上面的第三种解法,我们可以使用迭代的方式来实现,借助队列来实现。

/**
 * 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) {
        Set<Integer> set = new HashSet<Integer>();
        Queue<TreeNode> queue = new LinkedList<TreeNode>();
        queue.offer(root);
        while (!queue.isEmpty()) {
            TreeNode node = queue.poll();
            // 判断是否包含另外一个数字
            if (set.contains(k-node.val)) {
                return true;
            }
            set.add(node.val);
            // 将其左右子节点继续入队列
            if (node.left != null) {
                queue.offer(node.left);
            }
            if (node.right != null) {
                queue.offer(node.right);
            }
        }
        return false;
    }
}


06 小结

算法专题目前已日更超过四个月,算法题文章148+篇,公众号对话框回复【数据结构与算法】、【算法】、【数据结构】中的任一关键词,获取系列文章合集。

以上就是全部内容,如果大家有什么好的解法思路、建议或者其他问题,可以下方留言交流,点赞、留言、转发就是对我最大的回报和支持!

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 221,695评论 6 515
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 94,569评论 3 399
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 168,130评论 0 360
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 59,648评论 1 297
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 68,655评论 6 397
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 52,268评论 1 309
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,835评论 3 421
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,740评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 46,286评论 1 318
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 38,375评论 3 340
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 40,505评论 1 352
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 36,185评论 5 350
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,873评论 3 333
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 32,357评论 0 24
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,466评论 1 272
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,921评论 3 376
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 45,515评论 2 359

推荐阅读更多精彩内容