Python中怎么找出目標值的整數(shù)

題目要求:

給定一個整數(shù)數(shù)組 nums 和一個目標值 target，請你在該數(shù)組中找出和為目標值的那兩個整數(shù)，并返回他們的數(shù)組下標。

你可以假設每種輸入只會對應一個答案。但是，你不能重復利用這個數(shù)組中同樣的元素。

示例:

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

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

解題思路

給定數(shù)組arr,初始化字典dic,目標值target,a,b為數(shù)組的元素
如果存在這樣的關(guān)系a+b=target,如果a的下標為a_1,則可以得到dic[target-a]=a_1。
簡單數(shù)就是，當前的元素的下標和目標值減去當前元素獲取的值做一個映射，如果映射存在就可以找到這兩個數(shù)了。 

代碼

Python Version

from typing import List
class Solution:
    def twoSum(self, nums: List[int], target: int) -> List[int]:
        d =dict()
        for index,item in enumerate(nums):
            if item in d:
                return [d[item],index]
            else:
                d[target-item] =  index 

            #或者   
            # if target-item in d:
            #     return [d[target-item],index]
            # d[item] = index

if __name__ == "__main__":
    l:list = [2,7,11,15,12]
    target:int = 19
    s = Solution()
    print(s.twoSum(l,target))            

Go Version

package main
import "fmt"

func solution(nums []int,target int) []int{
    m:=make(map[int]int)
    for index,value := range nums {
        if w,ok:=m[value];ok{
            return []int{w,index}
        } else{
            m[target-value] = index
        }
    }
    return nil
}
func main() {
    nums:=[]int{2,7,11,15,12}
    fmt.Println("nums",nums)
    target:=19
    fmt.Println(solution(nums,target))

}