乐扣1:两数之和(tag:哈希)

前言:决定从今天刷LeeCode,选择了算法篇,因为之前觉得学了算法很长时间却没什么操练。可能刷起来比较慢,因为还有其他一些事情要做。总之,希望我的flag不要倒,多做题。
下面是乐扣的第一题。

题目

给定一个整数数组 nums 和一个目标值 target,请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,你不能重复利用这个数组中同样的元素。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

思路

第一眼看到题目想到的就是遍历数组,但是时间复杂度较高。然后想,能不能先把无序数组排序然后利用二分法查找,可是排序的时间复杂度已经和遍历数组一样了。这两种办法都不好,然后看到题目的tag是数组和hash,可能需要用哈希表的方法?其实看到哈希表有点懵,因为我是写go的,好像没遇见过go的哈希表?谷歌了一下才知道,原来go的map已经实现了hash的作用。那用map就可以了。

因为数组存的是值到下标的对应关系,如果给定一个target,对于某个数组中的值value,找到target-value的值对应下标的最快方法大概就是实现给定某个值快速找到它再数组中的下标,那么hash就是数组的值到这个值在数组中的下标的对应关系。

实现

初版代码

代码如下

func twoSum(nums []int, target int) []int {
    // value-当前值的结果到当前值的下标的映射,因为下标是从0开始的,所以为了避免map的歧义,在下标上加上1
    map1:=make(map[int]int)
    result:=make([]int,2)
    for idx:=0;idx<len(nums);idx++{
        value:=nums[idx]
        rest:=target-value
        map1[value]=idx+1
        if map1[rest]!=0{
            result[0]=idx
            result[1]=map1[rest]-1
            return result
        }
    }
    // 如果找不到
    return result
    
}

下标为0与map中key不存在返回0的冲突

然后提交说失败了,对于([2,4],6)这样的输入,答案为错误的[0,0]。我发现是if map1[rest]!=0那一行出错了,因为2对应的下标就是0,而我这里想判断的是如果map中不存在这个key的话,然后就改造程序:map存值到下标加1的对应关系,如下:

func twoSum(nums []int, target int) []int {
    // value-当前值的结果到当前值的下标的映射,因为下标是从0开始的,所以为了避免map的歧义,在下标上加上1
    map1:=make(map[int]int)
    result:=make([]int,2)
    for idx:=0;idx<len(nums);idx++{
        value:=nums[idx]
        rest:=target-value
        map1[value]=idx+1
        // 忽略重复的数据
        if target-value!=value && map1[rest]!=0{
            result[0]=idx
            result[1]=map1[rest]-1
            return result
        }
    }
    
    // 如果找不到
    return result
    
}

不同位置的相同值可以参与计算

运行显示,对于([3,3],6)这样的输入结果不对,原来是忽略重复数据那里。我判断应该是不能把数组中的用一个数(即下标相同的)计算两次,而不是两个不同下标的相同数不能相加。如果想实现这个的话,把判断target-value是否存在放在往map里添加值就可以了。
如下:

func twoSum(nums []int, target int) []int {
    // 当前值加上当前下标+1 到下标+1的映射,
    // 因为下标是从0开始的,所以为了避免map的歧义,在下标上加上1
    map1:=make(map[int]int)
    result:=make([]int,2)
    for idx:=0;idx<len(nums);idx++{
        value:=nums[idx]
        rest:=target-value
        if  map1[rest]!=0{
            // 忽略重复的数据,即自己和自己相加
            if idx+1==map1[rest]{
                continue
            }
            result[0]=idx
            result[1]=map1[rest]-1
            return result
        }
        map1[value]=idx+1
    }
    
    // 如果找不到
    return result
    
}

这次总算通过啦。

总结

  • 修改了几次,都是因为考虑不周引起。写程序测试时最好多考虑写异常路径。
  • 看了一下该问题的评论,用go写的人并不多,然后有一个地方可以优化,判断map中是否存在某值可以用如下代码,这样下标为0与map中key不存在返回0的冲突将不会存在。
if value, ok:=map1[key];ok{}
  • go的map效率很高,已经实现了哈希表的高效。再很多算法都可以用到这个数据结构。
  • 前期思考远比代码实现重要。

最终版

package main

import "fmt"

func twoSum(nums []int, target int) []int {
    // 当前值到下标的映射,
    map1:=make(map[int]int)
    result:=make([]int,2)
    for idx:=0;idx<len(nums);idx++{
        value:=nums[idx]
        rest:=target-value
        // 先判断有没有值存在再往map里面添加,
        // 避免输出类似([3,3],6)出错
        if  restvalue,ok:=map1[rest];ok{
            // 忽略重复的数据,即自己和自己相加
            if idx==restvalue{
                continue
            }
            result[0]=idx
            result[1]=restvalue
            return result
        }
        map1[value]=idx
        
    }
    
    // 如果找不到
    return result
    
}

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