leetcode_1. 两数之和

One minute read
LeetCode
面试 • 算法 • 数据结构

两数之和

题目

给定一个整数数组 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 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解答

解答一

暴力 复杂度为O(n2)

高低指针：低指针从前到后，高指针从低指针开始向后，遇到符合条件的退出

func twoSum(nums []int, target int) []int {
	if len(nums) == 0 {
		return []int{}
	}

	for k1, v1 := range nums {
		for k2, v2 := range nums[k1+1:] {
			if v1+v2 == target {
				return []int{k1, k1 + 1 + k2}
			}
		}
	}
	return []int{}
}

// 执行用时 :64 ms, 在所有 Go 提交中击败了38.45%的用户
// 内存消耗 :2.9 MB, 在所有 Go 提交中击败了100.00%的用户

解法二

遍历: O(n)

  • 如果 map 内含有 target - nums[i]，就返回 i 和 暂存的坐标
  • 使用map暂存值和坐标
func twoSum(nums []int, target int) []int {
	if len(nums) == 0 {
		return []int{}
	}

	m := make(map[int]int, len(nums)) // map[暂存值]坐标
    for k, v := range nums {
        if _, ok := m[target-v]; ok {
            return []int{m[target-v], k}
        }
        m[v] = k
    }
    return []int{}
}

// 执行用时 :4 ms, 在所有 Go 提交中击败了96.93%的用户
// 内存消耗 :3.4 MB, 在所有 Go 提交中击败了60.23%的用户

优化

创建map的时候，make使用len(nums)声明长度，会节约400K的内存

当 map 增长到容量上限的时候，如果再增加新的 key-value 对，map 的大小会自动加 1。 所以出于性能的考虑，对于大的 map 或者会快速扩张的 map，即使只是大概知道容量，也最好先标明。


分享：