1. 程式人生 > >陣列 兩數之和 go語言

陣列 兩數之和 go語言

給定一個整數陣列和一個目標值,找出陣列中和為目標值的兩個數。

你可以假設每個輸入只對應一種答案,且同樣的元素不能被重複利用。

示例:

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

因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
func twoSum(nums []int, target int) []int {
    kv := make(map[int]int, 0)
    for i := 0; i < len(nums); i++ {
        var com = target - nums[i]
        if k,ok:=kv[com];ok{
            return []int{k,i}
        }
        kv[nums[i]]=i
    }
    return nil
}