1. 程式人生 > >LeetCode Golang實現 1. 兩數之和

LeetCode Golang實現 1. 兩數之和

1. 兩數之和

給定一個整數陣列 nums 和一個目標值 target,請你在該陣列中找出和為目標值的那 兩個 整數,並返回他們的陣列下標。

你可以假設每種輸入只會對應一個答案。但是,你不能重複利用這個陣列中同樣的元素。

示例:

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

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

  

方法一: 暴力法:

leetCode 給出執行時間 60ms

 

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

  

方法二: 利用map, 藉助空間 降低時間複雜度

leetCode 給出執行時間 8ms

 

package main

import "fmt"

func main(){
	nums := []int{3,3}
	fmt.Println(twoSum(nums,6))
}


func twoSum(nums []int, target int) []int {
	numsMap := make(map[int]int)
	for i := 0; i < len(nums); i++ {
		numsMap[nums[i]] = i
	}
	for i := 0; i < len(nums); i++ {
		if _,ok := numsMap[target-nums[i]];ok && numsMap[target-nums[i]] != i{
			return []int{numsMap[target-nums[i]], i}
		}
	}

	return nil
}