2021-07-10 two-sum

https://leetcode-cn.com/problems/two-sum/

给定一个整数数组 nums 和一个整数目标值 target,请你在该数组中找出 和为目标值 target  的那 两个 整数,并返回它们的数组下标。

 你可以假设每种输入只会对应一个答案。但是,数组中同一个元素在答案里不能重复出现。

你可以按任意顺序返回答案。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

最朴素的解法就是循环:

func twoSum(nums []int, target int) []int {
    result := make([]int, 0, 2)
    length := len(nums)
    if length == 0 {
        return result
    }

    for i := 0; i < length-1; i++ {
        for j := i + 1; j < length; j++ {
            if nums[i]+nums[j] == target {
                result = append(result, i, j)
            }
        }
    }

    return result
}

这么样也可以通过,但是性能比较一般,时间复杂度为 O(n2)


image.png

我们也可以使用 map 来优化速度,这里需要注意,数组中元素可能是重复的,所以使用 map[int]int 没法满足需求,必须用 map[int][]int

代码如下:

func twoSum(nums []int, target int) []int {
    m := make(map[int][]int)

    for index, num := range nums {
        m[num] = append(m[num], index)
    }

    for i, num := range nums {
        left := target - num
        fmt.Println(i, num, left, m[left])
        if indexs, ok := m[left]; ok {
            // 从 indexs 里面找出不等于 i 的位置
            for _, index := range indexs {
                if index != i {
                    return []int{i, index}
                }
            }
        }
    }
    return []int{}
}

优化之后代码的性能提升巨大。


image.png

从 116ms -> 8ms 性能提升了几十倍,而且测试用例也全部通过。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容