Leetcode解题之路(golang版):1 两数之和(Two Sum)

1.两数之和

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

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

示例:

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

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

解法一:

空间复杂度:$O(1)$

时间复杂度:$O(n^2)$

//暴力循环两次
func twoSum(nums []int, target int) []int {
    a := make([]int, 2)
    for i := 0; i < len(nums) - 1; i++ {
        for j := i + 1; j < len(nums); j++ {
            if (nums[i] + nums[j] == target) {
                a[0] = i
                a[1] = j
            }
        }
    }
    return a
}

解法二:

空间复杂度:$O(n)$

时间复杂度:$O(n)$

//map记录 
//思路就是便利一次,但是每次都把当前数字存到map中;遇到新的数字就计算这个数字希望得到的数字是多少(target - nownumber),如果map中有,就说明找到了
//注意map的value存的是下标
func twoSum(nums []int, target int) []int {
    a := make(map[int]int, len(nums))
    b := make([]int, 2)
    for i := 0; i < len(nums); i++ {
        if v, ok := a[target - nums[i]]; ok {
            b[0] = i
            b[1] = v
            return b
        }
        a[nums[i]] = i
    }
    return b
}

你可能感兴趣的:(leetcode)