小明单身多年,有一天独自走着走着,看到路上成双成对的情侣,忽然想脱单!
于是决定去相个对象,他衣冠楚楚神采飞扬的来到婚姻介绍所。
红娘从档案柜翻呀翻,想找个合适的妹子给小明。可是翻了好久,都没有找到符合小明要求的女孩子。
第二天,终于来了个漂亮的女孩子。
故事讲到这里,你以为我是在讲程序员的相亲故事?
NO!NO!NO!
其实我要讲的还是我们程序员的老本行—算法编程题,话不多说,上酸菜!
给定一个整数数组 nums 和一个目标值 target,
请你在该数组中找出和为目标值的那 两个 整数,并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是,数组中同一个元素不能使用两遍。
示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/two-sum
拿到这道题,第一个想到的可能就是暴力算法,直接两层循环遍历一次不就OK了么,于是chua、chua、chua的就写出来了
解法一
// go代码
func twoSum(nums []int, target int) []int {
length := len(nums)
for i := 0; i < length; i++ {
for j := i + 1; j < length; j++ {
if nums[i]+nums[j] == target {
return []int{i, j}
}
}
}
return nil
}
可是看这算法的复杂度是相当的高,O(n*n)
解法二
暴力解法显然是不行的,我们怎么优化呢?
首先想到的办法就是对数组进行预处理(排序),在数组有序的情况下
可以使用头尾双指针遍历查找,数组排序的算法复杂度是O(nlogn),
遍历算法复杂度为O(n),整体的算法复杂度就降为O(nlogn)了,上正菜!
func twoSum2(nums []int, target int) []int {
// 快速排序算法O(nlogn)
sort.Ints(nums)
head := 0
tail := len(nums) - 1
for head < tail {
sum := nums[head] + nums[tail]
if sum == target {
return []int{head, tail}
}
if sum < target {
head++
} else {
tail--
}
}
return nil
}
算法看上去很美,其实这里有个非常严重的问题,就是你排序之后,原数组的元素顺序已经变了,所以得出的答案肯定是错误的。
解法三
于是,我们的主角小明上场了,小明凭借自己丰富的婚姻介绍所的经历,
想到其实这题跟自己去婚姻介绍所找妹子经历很像啊。小明和妹子们就相当于数组的元素,小明的求偶要求呢,就是我们题目的target,而婚姻介绍所是什么呢? 当然非我们的map莫属了,存储着所有善男信女的档案要求。上大菜!
func twoSum3(nums []int, target int) []int {
// 求偶登记册
targetMap := make(map[int]int)
for idx, val := range nums {
// 有没有人找我这样的暖男呢
targetIdx, ok := targetMap[val]
if ok {
// 哇,找到了
return []int{targetIdx, idx}
}
// 没有,没关系,我把我的目标要求登记下,下次有符合要求的及时通知我哦
targetMap[target-val] = idx
}
return nil
}
你看这道菜,是不是很香,是不是嘎嘣脆!
不过相比于前面两个,我们多了一个求偶登记册,也就是多了O(n)的空间复杂度,这也是空间换时间的经典算法呀。
最后祝小明顺利脱单,早日迎娶白富美!