【算法学习】少年，婚姻介绍所了解下

​小明单身多年，有一天独自走着走着，看到路上成双成对的情侣，忽然想脱单！

于是决定去相个对象，他衣冠楚楚神采飞扬的来到婚姻介绍所。

红娘从档案柜翻呀翻，想找个合适的妹子给小明。可是翻了好久，都没有找到符合小明要求的女孩子。

第二天，终于来了个漂亮的女孩子。

故事讲到这里，你以为我是在讲程序员的相亲故事？

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)的空间复杂度，这也是空间换时间的经典算法呀。

最后祝小明顺利脱单，早日迎娶白富美！