1. 两数之和

链接

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

你可以假设每种输入只会对应一个答案。但是，你不能重复利用这个数组中同样的元素。

示例:

给定 nums = [2, 7, 11, 15], target = 9

因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

题解

a + b = target

可以看成是

a = target - b

在map[整数]整数的序号中，可以查询到a的序号。这样就不用嵌套两个for循环了。

func twoSum(nums []int, target int) []int {
	// index 负责保存map[整数]整数的序列号
	index := make(map[int]int, len(nums))

	for i, b := range nums {
		// 通过查询map，获取a = target - b的序列号
		if j, ok := index[target-b]; ok {
			return []int{j, i}
			// 注意，顺序是j，i
		}
		// 把b和i的值，存入map
		index[b] = i
	}

	return nil
}