LeetCode之Two Sum IV - Input is a BST(Kotlin)

問題:
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.


方法:
與TwoSum的解法差不多,不過需要在其中加入樹的遍歷,深度優(yōu)先遍歷和廣度優(yōu)先遍歷都可以。把遍歷到的值存入到map中,如果在后面能夠找到與map中的值相加等于target的值就返回true,否則遍歷結(jié)束后返回false。

具體實(shí)現(xiàn):

class TwoSumIV {
    private val map = mutableMapOf<Int, Int>()
    private var target = 0

    class TreeNode(var `val`: Int = 0) {
        var left: TreeNode? = null
        var right: TreeNode? = null
    }

    /**
     * Definition for a binary tree node.
     * class TreeNode(var `val`: Int = 0) {
     *     var left: TreeNode? = null
     *     var right: TreeNode? = null
     * }
     */
    fun findTarget(root: TreeNode?, k: Int): Boolean {
        target = k
        return find(root)
    }

    private fun find(root: TreeNode?): Boolean {
        if (root == null) {
            return false
        }
        if (map[root.`val`] != null) {
            return true
        }
        map.put(target - root.`val`, 1)
        return find(root.left) || find(root.right)
    }
}

/**
 *      5
 *     / \
 *    3   6
 *   / \   \
 *  2   4   7
 */
fun main(args: Array<String>) {
    val root = TwoSumIV.TreeNode(5)
    root.left = TwoSumIV.TreeNode(3)
    root.right = TwoSumIV.TreeNode(6)
    (root.left)?.left = TwoSumIV.TreeNode(2)
    (root.left)?.right = TwoSumIV.TreeNode(4)
    (root.right)?.right = TwoSumIV.TreeNode(7)
    val twoSumIV = TwoSumIV()
    val result = twoSumIV.findTarget(root, 15)
    println("result $result")
}

有問題隨時(shí)溝通

具體代碼實(shí)現(xiàn)可以參考Github

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容