LeetCode-Kotlin-Array-1-Two Sum

Posted 黄毛火烧雪下

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了LeetCode-Kotlin-Array-1-Two Sum相关的知识,希望对你有一定的参考价值。

1. Two Sum

题目

Given an array of integers, return indices of the two numbers such that they add up to a specific target.

You may assume that each input would have exactly one solution, and you may not use the same element twice.

Example:

Given nums = [2, 7, 11, 15], target = 9,

Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1]

题目大意

在数组中找到 2 个数之和等于给定值的数字,结果返回 2 个数字在数组中的下标。

解题思路

这道题最优的做法时间复杂度是 O(n)。

顺序扫描数组,对每一个元素,在 map 中找能组合给定值的另一半数字,如果找到了,直接返回 2 个数字的下标即可。如果找不到,就把这个数字存入 map 中,等待扫到“另一半”数字的时候,再取出来返回结果。

代码

fun main() 
    var nums = intArrayOf(2, 7, 2, 15)
    var target = 9
    println(twoSum(nums, target).mapIndexed  index, value -> value )


fun twoSum(nums: IntArray, target: Int): IntArray 
    var result = mutableListOf<Int>()
    var mp= hashMapOf<Int,Int>()
    nums.forEachIndexed  index, value ->
        var middle=target-value
        if (mp.containsKey(middle))
            result.add(mp[middle]!!)
            result.add( index)
        else
            mp[value] =index
        
    
    return result.toIntArray()

[0, 1]

以上是关于LeetCode-Kotlin-Array-1-Two Sum的主要内容,如果未能解决你的问题,请参考以下文章