题目

给定一个整数数组 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)