LeetCode-0001 两数之和

题目

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

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

示例：

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

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

解答

方法 1：暴力破解

func twoSum(nums []int, target int) []int {
	if nums == nil {
		return nil
	}
	n := len(nums)
	if n < 2 {
	    return nil
	}
	var r [2]int
	for i := 1; i < n; i++ {
		for j := 0; j < i; j++ {
			if nums[j]+nums[i] == target {
				r[0], r[1] = j, i
				return r[:]
			}
		}
	}
	return nil
}

时间复杂度：O(n^2)

空间复杂度：常数

方法 2：运用 map

func twoSum(nums []int, target int) []int {
	if nums == nil {
		return nil
	}
	n := len(nums)
	if n < 2 {
	    return nil
	}
	m := map[int]int{}
	for i, y := range nums {
		m[y] = i
	}

	for i := 0; i < n; i++ {
		j, ok := m[target-nums[i]]
		if ok && i != j {
			return []int{i, j}
		}
	}
	return nil
}

时间复杂度（理想情形）：O(n)

空间复杂度：O(n)